feat:

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