HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

fix autobackend.py (#16801)

Browse source
This commit is contained in:
Mike 2024-10-09 21:58:50 +08:00 • committed by GitHub
parent 8154a27e9b
commit 28f31f14e8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/nn/autobackend.py
View file

@ -265,8 +265,8 @@ class AutoBackend(nn.Module):
if -1 in tuple(model.get_tensor_shape(name)): if -1 in tuple(model.get_tensor_shape(name)):
dynamic = True dynamic = True
context.set_input_shape(name, tuple(model.get_tensor_profile_shape(name, 0)[1])) context.set_input_shape(name, tuple(model.get_tensor_profile_shape(name, 0)[1]))
if dtype == np.float16: if dtype == np.float16:
fp16 = True fp16 = True
else: else:
output_names.append(name) output_names.append(name)
shape = tuple(context.get_tensor_shape(name)) shape = tuple(context.get_tensor_shape(name))