From 28f31f14e85449256a4b05934ba356a11315aaf7 Mon Sep 17 00:00:00 2001 From: Mike <60082702+lmk123568@users.noreply.github.com> Date: Wed, 9 Oct 2024 21:58:50 +0800 Subject: [PATCH] fix autobackend.py (#16801) --- ultralytics/nn/autobackend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index 725c6ba48a..22d4e6b0e7 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))