快捷方式

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][源代码]

返回已注册模型名称的列表。

参数:
  • module (ModuleType, 可选) – 我们要从中提取可用模型的模块。

  • include (strIterable[str], 可选) – 用于从所有模型集中包含模型的过滤器。过滤器传递给 fnmatch 以匹配 Unix shell 样式通配符。在有多个过滤器的情况下,结果是各个过滤器的并集。

  • exclude (strIterable[str], 可选) – 在 include_filters 之后应用的过滤器以删除模型。过滤器传递给 fnmatch 以匹配 Unix shell 样式通配符。在有多个过滤器的情况下,结果是删除与任何单个过滤器匹配的所有模型。

返回值:

可用模型名称的列表。

返回类型:

models (list)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源