Hi all,
Just curious if there is a better way to list all classification models in torchvision
besides something like torchvision.models.__dict__.keys()
I am trying to find something similar to pytorch-image-models
you can do timm.list_models()
to get a list of all available models.
You used to be able to query all segmentation models via something like __all__ = ['fcn_resnet50', 'fcn_resnet101', 'deeplabv3_resnet50', 'deeplabv3_resnet101']
from vision/segmentation.py at v0.6.0 · pytorch/vision · GitHub, but I see that was removed in v0.12.0
Thanks!