Create README.md

README.md

@@ -0,0 +1,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",
+    ),