Pytorch 入门实战 手搓自己的第一个网络
写在前面 最近老想用Pytorch来搓点网络了。 第一个例子:简单的CNN图像分类 一段一段代码来分析 import torch import torchvision import torchvision.transforms as transforms 导入必要的库,不扯淡 transform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) batch_size = 4 trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True,…