缘由
安装PyTorch后就想尝试能不能跑着玩玩,就在官网看到了一个训练图像分类器的小教程;
纯CPU运算
import torch
import torchvision
import torchvision.transforms as transforms
from datetime import datetime
transform = transforms.Compose(
[transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
trainset = torchvision.datasets.CI......