get_model_weights¶ torchvision.models.get_model_weights(name: Union[Callable, str]) → Type[WeightsEnum][源代码]¶ 返回与给定模型关联的权重枚举类。 参数: name (callable 或 str) – 模型构建器函数或其注册名称。 返回: 与模型关联的权重枚举类。 返回类型: weights_enum (WeightsEnum)
