File size: 635 Bytes
69f6592 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
["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",
), |