Updated batch size to match our laptop GPUs

MedicalVR · Jan 20, 2023 · c9c7141 · c9c7141
1 parent 347f2e2
commit c9c7141
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config.py b/config.py
@@ -35,7 +35,7 @@ class TrainSegmentatorConfig:
     split: str = None
 
     # Dataloader
-    batch_size: int = 4
+    batch_size: int = 2
     num_workers: int = 10
 
     # Model