config.json

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