Update outerRaceModel.py

outerRaceModel.py

@@ -56,7 +56,7 @@ from dataclasses import dataclass
 
 @dataclass
 class TrainingConfig:
-    image_size = 32  # the generated image resolution
+    image_size = 256  # the generated image resolution
     train_batch_size = 10
     eval_batch_size = 16  # how many images to sample during evaluation
     num_epochs = 250
@@ -262,7 +262,7 @@ def evaluate(config, epoch, pipeline):
             os.makedirs(test_dir)
     
         for i, image in enumerate(images):
-            image.save(f"{test_dir}/{(i+(k*16)):04d}.png")
+            image.save(f"{test_dir}/{(i+((k-1)*16)):04d}.png")
 
         config.seed = random.randint(1, 1000)