快捷方式

get_model

torchvision.models.get_model(name: str, **config: Any) Module[來源]

取得模型名稱和配置,並返回一個實例化的模型。

參數:
  • name (str) – 模型註冊的名稱。

  • **config (Any) – 傳遞給模型建構方法的參數。

返回:

初始化的模型。

返回類型:

模型 (nn.Module)

使用 get_model 的範例

Transforms v2:端到端物件偵測/分割範例

Transforms v2:端到端物件偵測/分割範例

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得解答

檢視資源