From 03c85991e3bcdf55c30b721c95507dc25e27bc15 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 26 Dec 2022 16:19:55 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- timm_image.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/timm_image.py b/timm_image.py index b9e03cd..8668525 100644 --- a/timm_image.py +++ b/timm_image.py @@ -189,7 +189,8 @@ class TimmImage(NNOperator): @staticmethod def supported_model_names(format: str = None): - assert timm.__version__ == '0.6.12', 'The model lists are tested with timm==0.6.12.' + if timm.__version__ != '0.6.12': + log.warning('Please note that the model list is tested with timm==0.6.12, please check your timm version.') full_list = list(set(timm.list_models(pretrained=True)) - set([ 'coat_mini', 'coat_tiny',