From 09959f1e93eed7751a0ece03e850b6e35cd252f8 Mon Sep 17 00:00:00 2001 From: zzw <1464003642@qq.com> Date: Wed, 19 Aug 2026 17:53:35 +0800 Subject: [PATCH] feat: --- train_yolov8.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/train_yolov8.py b/train_yolov8.py index eb45630..54f96fa 100644 --- a/train_yolov8.py +++ b/train_yolov8.py @@ -8,16 +8,15 @@ def parse_args(): parser = ArgumentParser(description="Train YOLOv8 on this dataset.") 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=640) - parser.add_argument("--batch", type=int, default=8, help="Use -1 for automatic batch size.") + 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=4) + parser.add_argument("--workers", type=int, default=6) 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.") return parser.parse_args() - def main(): args = parse_args() root = Path(__file__).resolve().parent