feat: run

2026-08-19 17:58:30 +08:00
parent bff7bde701
commit 534e7d410e
1 changed files with 1 additions and 1 deletions
@@ -9,7 +9,7 @@ def parse_args():
    parser.add_argument("--model", default="yolov8n.pt", help="Model or checkpoint path.")
    parser.add_argument("--epochs", type=int, default=1)
    parser.add_argument("--imgsz", type=int, default=1024)
-    parser.add_argument("--batch", type=int, default=8, help="Use -1 for automatic batch size.")
+    parser.add_argument("--batch", type=int, default=16, help="Use -1 for automatic batch size.")
    parser.add_argument("--device", default="0", help="CUDA device such as 0, or cpu.")
    parser.add_argument("--workers", type=int, default=4)
    parser.add_argument("--patience", type=int, default=30)