Get All Available Models in Torchvision

Torchvision is a popular computer vision library in PyTorch that offers various pre-trained models. These models can be a convenient starting point for building computer vision applications. By understanding the available models, we can make informed decisions, select the most suitable model for our applications. This tutorial shows how to get all available models in Torchvision.

Code

In the following code, we retrieve and print the names of all available models in Torchvision using the list_models function from the torchvision.models package.

from torchvision.models import list_models

for name in list_models():
    print(name)

Here's an example of the output you might see when running the code snippet:

alexnet
convnext_base
convnext_large
convnext_small
convnext_tiny
...