I’m trying to learn to use PyTorch and the first thing I wanted to do was MNIST predictor. I got pretty good results (something like 99%) but I want to use some tricks which Jeremy have been taught. The code is below so can someone explain to me how I can rotate the numbers a little to get more training data (aka data augmentation).
MNIST dataset
train_dataset = torchvision.datasets.MNIST(root='../../data/',
train=True,
transform=transforms.ToTensor(),
download=True)
test_dataset = torchvision.datasets.MNIST(root='../../data/',
train=False,
transform=transforms.ToTensor())
Data loader
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
batch_size=batch_size,
shuffle=True)
test_loader = torch.utils.data.DataLoader(dataset=test_dataset,
batch_size=batch_size,
shuffle=False)