NaviGlass/Yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

关闭AMP

Browse Source
This commit is contained in:
Kevin Wong
2026-01-05 14:06:53 +08:00
parent fec2523b5c
commit 861772950f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_merged.py
View File

@@ -21,7 +21,7 @@ results = model.train(
cache='ram', # 192GB 内存全量缓存 cache='ram', # 192GB 内存全量缓存
optimizer='AdamW', optimizer='AdamW',
close_mosaic=15, # 最后15轮关闭mosaic close_mosaic=15, # 最后15轮关闭mosaic
amp=True, # 混合精度 amp=False, # 混合精度 (NaN时关闭)
patience=30, # 早停耐心值 patience=30, # 早停耐心值
project="blind_guide_project", project="blind_guide_project",
name="yolo11l_blind_v2" # 版本2 name="yolo11l_blind_v2" # 版本2