List models that support features_only=True feature extraction

[isaaccorley]

I have a use case for a library I'm working on TorchSeg where I want to list all available timm models that support features_only=True.

I'm currently looping over all possible models and catching the exception if it doesn't support it, but I was curious if there was already an available list in timm?

import timm

models = timm.list_models()
works, fails = [], []
for model in timm.list_models():
    try:
        m = timm.create_model(model, pretrained=False, features_only=True)
        works.append(model)
    except RuntimeError:
        fails.append(model)

[rwightman]

There's an open issue for that one #1824 ... will make a comment there about last thoughts I had on this...

[isaaccorley]

Thanks, my initial search didn't catch that. I'll close this and follow up there.