跑了一个epoch,正确率为:
import torch
import torch.nn as nn
import torchvision.datasets as dsets
import torchvision.transforms as transforms
from torch.autograd import Variable
#Image Preprocessing
transform = transforms.Compose([
transforms.Scale(40),
transforms.RandomHorizontalFlip(),
transforms.RandomCrop(32),
transforms.ToTensor()
])
#CIFAR-10 Dataset
train_dataset = dsets.CIFAR10(
root='./data',
train=True,
transform = transform,
download=True
)
test_dataset = dsets.CIFAR10(
root='./data',
train=False,
transform = transform,
download=True
)
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
batch_size=50,