diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index 725c6ba4..22d4e6b0 100644 --- a/ultralytics/nn/autobackend.py +++ b/ultralytics/nn/autobackend.py @@ -265,8 +265,8 @@ class AutoBackend(nn.Module): if -1 in tuple(model.get_tensor_shape(name)): dynamic = True context.set_input_shape(name, tuple(model.get_tensor_profile_shape(name, 0)[1])) - if dtype == np.float16: - fp16 = True + if dtype == np.float16: + fp16 = True else: output_names.append(name) shape = tuple(context.get_tensor_shape(name))