get_model_weights¶ torchvision.models.get_model_weights(name: Union[Callable, str]) → type[torchvision.models._api.WeightsEnum][source]¶ 返回與給定模型關聯的權重列舉類。 引數: name (callable 或 str) – 模型構建函式或註冊它的名稱。 返回: 與模型關聯的權重列舉類。 返回型別: weights_enum (WeightsEnum)
