{
  "_name_or_path": "/root/CS492-Team28-Diffusion-Project-3DVolume/result/112611/checkpoint-11250",
  "architectures": [
    "UNet3D"
  ],
  "attn_resolutions": [
    4,
    8,
    16
  ],
  "ch": 64,
  "ch_mult": [
    1,
    2,
    4,
    8,
    8
  ],
  "dropout": 0.1,
  "image_size": 64,
  "in_channels": 1,
  "model_type": "unet3d",
  "num_classes": 3,
  "num_diffusion_timesteps": 1000,
  "num_res_blocks": 2,
  "out_ch": 1,
  "resamp_with_conv": true,
  "torch_dtype": "float32",
  "transformers_version": "4.45.1"
}