File size: 649 Bytes
485c2ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
{
"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"
}
|