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

feat: test run

Browse Source
This commit is contained in:
zzw
2026-08-19 18:00:53 +08:00
parent b920fc7646
commit de53b945bc
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=9, 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.")
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)