From 8c66702529a4d99699c5bed8ee08e67d4b7d59f3 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 26 Dec 2022 16:26:05 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- timm_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm_image.py b/timm_image.py index 8668525..bc2ff0e 100644 --- a/timm_image.py +++ b/timm_image.py @@ -256,7 +256,7 @@ class TimmImage(NNOperator): 'volo_d5_448', 'volo_d5_512' ] - assert set(to_remove).issubset(set(full_list)) + # assert set(to_remove).issubset(set(full_list)) model_list = list(set(full_list) - set(to_remove)) # todo: elif format == 'torchscript': # todo: elif format == 'tensorrt'