Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
2 additions and
1 deletions
-
timm_image.py
|
|
@ -189,7 +189,8 @@ class TimmImage(NNOperator): |
|
|
|
|
|
|
|
|
@staticmethod |
|
|
@staticmethod |
|
|
def supported_model_names(format: str = None): |
|
|
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([ |
|
|
full_list = list(set(timm.list_models(pretrained=True)) - set([ |
|
|
'coat_mini', |
|
|
'coat_mini', |
|
|
'coat_tiny', |
|
|
'coat_tiny', |
|
|
|
