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.")