田运杰 преди 10 месеца
родител
ревизия
4d00e60c70
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  1. 2 1
      ultralytics/utils/checks.py

+ 2 - 1
ultralytics/utils/checks.py

@@ -702,7 +702,8 @@ def check_amp(model):
     try:
         from ultralytics import YOLO
 
-        assert amp_allclose(YOLO("yolo11n.pt"), im)
+        # assert amp_allclose(YOLO("yolo11n.pt"), im)
+        assert amp_allclose(YOLO("yolov12n.pt"), im)
         LOGGER.info(f"{prefix}checks passed ✅")
     except ConnectionError:
         LOGGER.warning(