Shortcuts

list_models

torchvision.models.list_models(module: Optional[module] = None, include: Optional[Union[Iterable[str], str]] = None, exclude: Optional[Union[Iterable[str], str]] = None) List[str][source]

Returns a list with the names of registered models.

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

  • include (str or Iterable[str], optional) – Filter(s) for including the models from the set of all models. Filters are passed to fnmatch to match Unix shell-style wildcards. In case of many filters, the results is the union of individual filters.

  • exclude (str or Iterable[str], optional) – Filter(s) applied after include_filters to remove models. Filter are passed to fnmatch to match Unix shell-style wildcards. In case of many filters, the results is removal of all the models that match any individual filter.

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