Fixed cifar transform
@@ -169,12 +169,12 @@ def main(hparams):
])
trainset = torchvision.datasets.CIFAR10(root=data_path, train=True,
- download=True, transform=transform)
+ download=True, transform=transform_train)
trainloader = torch.utils.data.DataLoader(trainset, batch_size = batch_size_train,
shuffle=True, num_workers=2)
testset = torchvision.datasets.CIFAR10(root=data_path, train=False,
- download=True, transform=transform)
+ download=True, transform=transform_test)
testloader = torch.utils.data.DataLoader(testset, batch_size = batch_size_test,
shuffle=False, num_workers=2)
