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 (str 或 Iterable[str], 可選) – 用於從所有模型集合中包含模型的過濾器。過濾器會被傳遞給 fnmatch 以匹配 Unix shell 風格的萬用字元。如果存在多個過濾器，結果將是單個過濾器的並集。

• exclude (str 或 Iterable[str], 可選) – 在 include_filters 之後應用的過濾器，用於移除模型。過濾器會被傳遞給 fnmatch 以匹配 Unix shell 風格的萬用字元。如果存在多個過濾器，結果將是移除所有匹配任一單個過濾器的模型。

返回值：

可用模型的名稱列表。

返回型別：

models (list)