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__':