Update innerRace.py

innerRace.py

@@ -56,15 +56,15 @@ 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 = 2  # how many images to sample during evaluation
+    eval_batch_size = 16  # how many images to sample during evaluation
     num_epochs = 250
     gradient_accumulation_steps = 1
     learning_rate = 1e-4
     lr_warmup_steps = 500
-    save_image_epochs = 2
-    save_model_epochs = 2
+    save_image_epochs = 100
+    save_model_epochs = 100
     mixed_precision = "fp16"  # `no` for float32, `fp16` for automatic mixed precision
     output_dir = "7inchInnerRace1730"  # the model name locally and on the HF Hub