From f40a4fd06efce85adf4655a4bb3d0a0c150df1ef Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Sat, 2 Apr 2022 11:16:19 +0800
Subject: [PATCH] Align output name

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 timm_image.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/timm_image.py b/timm_image.py
index 32ed98d..1c2a70f 100644
--- 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__':