YULI/xunlian
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: nc

Browse Source
This commit is contained in:
zzw
2026-08-19 16:47:59 +08:00
parent 7fde28dacb
commit 5653153120
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_yolov8.py
+1 -1
View File
@@ -7,7 +7,7 @@ from ultralytics import YOLO
def parse_args(): def parse_args():
parser = ArgumentParser(description="Train YOLOv8 on this dataset.") parser = ArgumentParser(description="Train YOLOv8 on this dataset.")
parser.add_argument("--model", default="yolov8n.pt", help="Model or checkpoint path.") parser.add_argument("--model", default="yolov8n.pt", help="Model or checkpoint path.")
parser.add_argument("--epochs", type=int, default=100) parser.add_argument("--epochs", type=int, default=1)
parser.add_argument("--imgsz", type=int, default=640) 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("--batch", type=int, default=8, help="Use -1 for automatic batch size.")
parser.add_argument("--device", default="0", help="CUDA device such as 0, or cpu.") parser.add_argument("--device", default="0", help="CUDA device such as 0, or cpu.")