tuklu
/

pothole_detection

Jisnu Kalita Claude Sonnet 4.6 commited on 6 days ago

Commit

c79a620

1 Parent(s): 6124b76

increase batch to 128 and imgsz to 1280 for full VRAM utilization

Files changed (1) hide show

main.py CHANGED Viewed

@@ -13,8 +13,8 @@ def train():
     results = model.train(
         data=str(DATASET_YAML),
         epochs=200,
-        imgsz=640,
-        batch=16,
         project=str(RUNS_DIR),
         name="pothole",
         exist_ok=True,

     results = model.train(
         data=str(DATASET_YAML),
         epochs=200,
+        imgsz=1280,
+        batch=128,
         project=str(RUNS_DIR),
         name="pothole",
         exist_ok=True,