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