|
|
@ -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
|
|
|
|