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