get_model_weights

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

返回与给定模型关联的权重枚举类。

参数：

name (callable 或 str) – 模型构建函数或其注册的名称。

返回值：

与模型关联的权重枚举类。

返回类型：

weights_enum (WeightsEnum)