| { | |
| "ampmode": "16-mixed", | |
| "architectures": [ | |
| "DiffAE" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "DiffAEConfig.DiffAEConfig", | |
| "AutoModel": "DiffAE.DiffAE" | |
| }, | |
| "batch_size": 9, | |
| "data_name": "ukbb", | |
| "diffusion_type": "beatgans", | |
| "grey2RGB": -1, | |
| "in_channels": 1, | |
| "input_shape": [ | |
| 50, | |
| 128, | |
| 128 | |
| ], | |
| "is3D": true, | |
| "latent_dim": 128, | |
| "lr": 0.0001, | |
| "model_type": "DiffAE", | |
| "net_ch": 32, | |
| "out_channels": 1, | |
| "sample_every_batches": 1000, | |
| "sample_size": 4, | |
| "seed": 1701, | |
| "test_ema": true, | |
| "test_emb_only": true, | |
| "test_with_TEval": true, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.44.2" | |
| } | |