| { |
| "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" |
| } |
|
|
