Shortcuts

list_models

torchvision.models.list_models(module: Optional[module] = None) List[str][source]

Returns a list with the names of registered models.

Warning

The function is in Beta stage, and backward compatibility is not guaranteed.

Parameters:

module (ModuleType, optional) – The module from which we want to extract the available models.

Returns:

A list with the names of available models.

Return type:

models (list)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources