soumickmj's picture
Upload DiffAE
2cb251e verified
raw
history blame contribute delete
649 Bytes
{
"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"
}