Shortcuts

get_weight

torchvision.models.get_weight(name: str) WeightsEnum[source]

通过全名获取权重枚举值。示例:“ResNet50_Weights.IMAGENET1K_V1”

Parameters:

name (str) – 权重枚举条目的名称。

Returns:

请求的权重枚举。

Return type:

权重枚举