From a04940fb30ebbe1ee828ac177f66bd7b0cb69cf4 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 25 Mar 2022 17:51:10 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- timm_image.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/timm_image.py b/timm_image.py index 09361e7..fe2cc97 100644 --- a/timm_image.py +++ b/timm_image.py @@ -30,10 +30,9 @@ from timm.models.factory import create_model import warnings warnings.filterwarnings('ignore') - - log = logging.getLogger() + @register(output_schema=['vec']) class TimmImage(NNOperator): """