田运杰 hace 10 meses
padre
commit
4d00e60c70
Se han modificado 1 ficheros con 2 adiciones y 1 borrados
  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(