Align output name

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · f40a4fd06e
1 changed files with 2 additions and 2 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -70,8 +70,8 @@ class TimmImage(NNOperator):
            features = global_pool(features)

        features = features.to('cpu')
-        feature_vector = features.flatten().detach().numpy()
-        return feature_vector
+        vec = features.flatten().detach().numpy()
+        return vec


 # if __name__ == '__main__':