| dataset_params: | |
| im_path: 'data/mnist/train/images' | |
| im_test_path: 'data/mnist/test/images' | |
| canny_im_size: 28 | |
| diffusion_params: | |
| num_timesteps : 1000 | |
| beta_start : 0.0001 | |
| beta_end : 0.02 | |
| model_params: | |
| im_channels : 1 | |
| im_size : 28 | |
| hint_channels : 3 | |
| down_channels : [32, 64, 128, 256] | |
| mid_channels : [256, 256, 128] | |
| down_sample : [True, True, False] | |
| time_emb_dim : 128 | |
| num_down_layers : 2 | |
| num_mid_layers : 2 | |
| num_up_layers : 2 | |
| num_heads : 4 | |
| train_params: | |
| task_name: 'mnist' | |
| batch_size: 64 | |
| num_epochs: 40 | |
| controlnet_epochs : 1 | |
| num_samples : 25 | |
| num_grid_rows : 5 | |
| save_epoch: 2 | |
| ddpm_lr: 0.0001 | |
| controlnet_lr: 0.0001 | |
| ddpm_ckpt_name: 'ddpm_ckpt.pth' | |
| controlnet_ckpt_name: 'ddpm_controlnet_ckpt.pth' | |