| ["0102", "0295", "0071", "0174", "0106", "0116", "0065"] | |
| eval_batch_size = 25 # how many images to sample during evaluation | |
| num_epochs = 10000 | |
| gradient_accumulation_steps = 1 | |
| learning_rate = 5e-5 | |
| transforms.Resize((config.image_size, config.image_size)), | |
| transforms.RandomHorizontalFlip(), | |
| transforms.ToTensor(), | |
| transforms.Normalize([0.5], [0.5]), | |
| "DownBlock2D", # a regular ResNet downsampling block | |
| "DownBlock2D", | |
| "DownBlock2D", | |
| "DownBlock2D", | |
| "AttnDownBlock2D", # a ResNet downsampling block with spatial self-attention | |
| "DownBlock2D", | |
| ), |