PyTorch 1.0 中文文档：torchvision.models

ApacheCN_飞龙

发布于 2022-05-07 14:24:58

1770

发布于 2022-05-07 14:24:58

文章被收录于专栏：信数据得永生信数据得永生

译者：BXuan694

models子包定义了以下模型架构：

你可以通过调用以下构造函数构造随机权重的模型：

import torchvision.models as models
resnet18 = models.resnet18()
alexnet = models.alexnet()
vgg16 = models.vgg16()
squeezenet = models.squeezenet1_0()
densenet = models.densenet161()
inception = models.inception_v3()

我们在torch.utils.model_zoo中提供了预训练模型。预训练模型可以通过传递参数pretrained=True构造：

import torchvision.models as models
resnet18 = models.resnet18(pretrained=True)
alexnet = models.alexnet(pretrained=True)
squeezenet = models.squeezenet1_0(pretrained=True)
vgg16 = models.vgg16(pretrained=True)
densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True)