Skip to content

What is the selected model when just the base model's name is written? #1896

Answered by rwightman
mjack3 asked this question in Q&A
Discussion options

You must be logged in to vote

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)

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by mjack3
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants