| { | |
| "image_size": 256, | |
| "num_classes": 10, | |
| "latent_channels": 4, | |
| "latent_size": 32, | |
| "text_raw_dim": 2048, | |
| "bottleneck_dim": 256, | |
| "tower_dim": 256, | |
| "tower_depth": 2, | |
| "num_heads": 8, | |
| "geometric_types": [ | |
| "cantor", | |
| "beatrix" | |
| ], | |
| "conv_types": [ | |
| "wide_resnet", | |
| "squeeze_excite" | |
| ], | |
| "conv_spatial_size": 8, | |
| "manifold_dim": 512, | |
| "fingerprint_dim": 64, | |
| "num_flow_steps": 50 | |
| } |