get_model_weights

torchvision.models.get_model_weights(name: Union[Callable, str]) → Type[WeightsEnum]

傳回與指定模型相關聯的權重列舉類別。

參數:

name (可呼叫物件 或 字串) – 模型建構器函式或其註冊名稱。

傳回:

與模型相關聯的權重列舉類別。

傳回類型:

weights_enum (WeightsEnum)