Training for longer duration

GeroVanMi · May 15, 2024 · 5787951 · 5787951
1 parent 425aa1c
commit 5787951
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/src/configurations/TrainConfig.py b/training/src/configurations/TrainConfig.py
@@ -7,5 +7,5 @@
 class TrainConfig(Configuration):
     train_batch_size = 96
     eval_batch_size = 8
-    num_epochs = 30
+    num_epochs = 60
     num_images = 7351