Update detector.py

umyelab · Nov 7, 2024 · 81179c6 · 81179c6
1 parent 9eebfff
commit 81179c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/LabGym/detector.py b/LabGym/detector.py
@@ -95,7 +95,7 @@ def train(self,path_to_annotation,path_to_trainingimages,path_to_detector,iterat
 		cfg.SOLVER.GAMMA=0.5
 		cfg.SOLVER.IMS_PER_BATCH=4
 		cfg.MODEL.DEVICE=self.device
-		cfg.SOLVER.CHECKPOINT_PERIOD=10000000
+		cfg.SOLVER.CHECKPOINT_PERIOD=10000000000
 		cfg.INPUT.MIN_SIZE_TEST=int(inference_size)
 		cfg.INPUT.MAX_SIZE_TEST=int(inference_size)
 		cfg.INPUT.MIN_SIZE_TRAIN=(int(inference_size),)