diff --git a/train_merged.py b/train_merged.py index dac10e1..b9ef8b5 100644 --- 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