关闭AMP

2026-01-05 14:06:53 +08:00
parent fec2523b5c
commit 861772950f
1 changed files with 1 additions and 1 deletions
--- a/train_merged.py
+++ b/train_merged.py
@@ -21,7 +21,7 @@ results = model.train(
    cache='ram',          # 192GB 内存全量缓存
    optimizer='AdamW',
    close_mosaic=15,      # 最后15轮关闭mosaic
-    amp=True,             # 混合精度
+    amp=False,            # 混合精度 (NaN时关闭)
    patience=30,          # 早停耐心值
    project="blind_guide_project",
    name="yolo11l_blind_v2"  # 版本2