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',