logo
Browse Source

Align output name

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
Jael Gu 3 years ago
parent
commit
f40a4fd06e
  1. 4
      timm_image.py

4
timm_image.py

@ -70,8 +70,8 @@ class TimmImage(NNOperator):
features = global_pool(features) features = global_pool(features)
features = features.to('cpu') features = features.to('cpu')
feature_vector = features.flatten().detach().numpy()
return feature_vector
vec = features.flatten().detach().numpy()
return vec
# if __name__ == '__main__': # if __name__ == '__main__':

Loading…
Cancel
Save