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)
