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

feat: run

Browse Source
This commit is contained in:
zzw
2026-08-19 17:59:50 +08:00
parent 534e7d410e
commit b920fc7646
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
@@ -9,7 +9,7 @@ def parse_args():
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=1) parser.add_argument("--epochs", type=int, default=1)
parser.add_argument("--imgsz", type=int, default=1024) parser.add_argument("--imgsz", type=int, default=1024)
parser.add_argument("--batch", type=int, default=16, help="Use -1 for automatic batch size.") parser.add_argument("--batch", type=int, default=9, 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.")
parser.add_argument("--workers", type=int, default=4) parser.add_argument("--workers", type=int, default=4)
parser.add_argument("--patience", type=int, default=30) parser.add_argument("--patience", type=int, default=30)