PyTorch:torchvision.models

torchvision.models模块的 子模块中包含以下模型结构。AlexNetVGGResNetSqueezeNet你可以使用随机初始化的权重来创建这些模型。ResNet variants和AlexNet,我们也提供了预训练(pre-trained)的模型。ImageNet 1-crop error rates (224×224)NetworkTop-1 errorTop-5 errorRe