分类 Pytorch 下的文章

torchvision 是PyTorch中专门用来处理图像的库。这个包中有四个大类。torchvision.datasets 用来进行数据加载的,PyTorch团队在这个包中帮我们提前处理好了很多很多图片数据集。torchvision.models 提供已经训练好的模型,让我们可以加载之后,直接使用。我们可以直接使用如下代码来快速创建一个权重随机初始化的模型:import torchvision.models as modelsresnet18 = models.resnet18()alexnet = models.alexnet()squeezenet = models.squeezen……