disabled AMP check (not supported anyway)

single_channel
Samuel Dobron 6 months ago
parent d92dce3a46
commit 9e42d05e01

@ -448,6 +448,7 @@ def check_amp(model):
Returns: Returns:
(bool): Returns True if the AMP functionality works correctly with YOLOv8 model, else False. (bool): Returns True if the AMP functionality works correctly with YOLOv8 model, else False.
""" """
return False
device = next(model.parameters()).device # get model device device = next(model.parameters()).device # get model device
if device.type in ('cpu', 'mps'): if device.type in ('cpu', 'mps'):
return False # AMP only used on CUDA devices return False # AMP only used on CUDA devices

Loading…
Cancel
Save