From 9e42d05e018b6974b9429df464e2c885f2b8205a Mon Sep 17 00:00:00 2001 From: Samuel Dobron Date: Sat, 1 Jun 2024 14:01:16 +0000 Subject: [PATCH] disabled AMP check (not supported anyway) --- ultralytics/utils/checks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ultralytics/utils/checks.py b/ultralytics/utils/checks.py index 505abcd..68d66de 100644 --- a/ultralytics/utils/checks.py +++ b/ultralytics/utils/checks.py @@ -448,6 +448,7 @@ def check_amp(model): Returns: (bool): Returns True if the AMP functionality works correctly with YOLOv8 model, else False. """ + return False device = next(model.parameters()).device # get model device if device.type in ('cpu', 'mps'): return False # AMP only used on CUDA devices