From 78e50bdf09798b2d276bf10dc937448c531c3a73 Mon Sep 17 00:00:00 2001 From: zzw <1464003642@qq.com> Date: Wed, 19 Aug 2026 17:55:23 +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 54f96fa..c389076 100644 --- a/train_yolov8.py +++ b/train_yolov8.py @@ -11,7 +11,7 @@ def parse_args(): parser.add_argument("--imgsz", type=int, default=1024) parser.add_argument("--batch", type=int, default=-1, 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=6) + parser.add_argument("--workers", type=int, default=4) parser.add_argument("--patience", type=int, default=30) parser.add_argument("--name", default="yolov8n_acne") parser.add_argument("--resume", action="store_true", help="Resume from --model checkpoint.")