-
Hello, What model from the list below is chosen if
The list above is after writing Thanks |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
The priority order is currently lost in the list modules fn, The official order for a given model can be found via Above is obviously not a stable API. The order is based on the order in the default_cfgs (https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/resnet.py#L594) .. the first 'tag' for a given architecture (base name) is what will be used if you don't specify the tag It could be possible to tweak list_models to keep the tag priority order, would need to re-write a chunk to not use a set (so order isn't lost) |
Beta Was this translation helpful? Give feedback.
The priority order is currently lost in the list modules fn,
The official order for a given model can be found via
timm.models._registry._model_with_tags['resnet50']
Above is obviously not a stable API. The order is based on the order in the default_cfgs (https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/resnet.py#L594) .. the first 'tag' for a given architecture (base name) is what will be used if you don't specify the tag
It could be possible to tweak list_models to keep the tag priority order, would need to re-write a chunk to not use a set (so order isn't lost)