From 534e7d410ec56b596a5f989e9c96f2a2a2daa9e7 Mon Sep 17 00:00:00 2001 From: zzw <1464003642@qq.com> Date: Wed, 19 Aug 2026 17:58:30 +0800 Subject: [PATCH] feat: run --- train_yolov8.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_yolov8.py b/train_yolov8.py index cdfcb5d..9461359 100644 --- a/train_yolov8.py +++ b/train_yolov8.py @@ -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)