From 2f06ac98982323c3775faba1f5f64b52b5586b70 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 15 Mar 2022 19:45:40 +0800 Subject: Remove data augmentation on test set --- supervised/baseline.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'supervised') diff --git a/supervised/baseline.py b/supervised/baseline.py index 6a83f72..9a83079 100644 --- a/supervised/baseline.py +++ b/supervised/baseline.py @@ -74,15 +74,20 @@ def get_color_distortion(s=1.0): return color_distort -transform = transforms.Compose([ +train_transform = transforms.Compose([ transforms.RandomResizedCrop(32, interpolation=InterpolationMode.BICUBIC), transforms.RandomHorizontalFlip(0.5), get_color_distortion(0.5), transforms.ToTensor() ]) -train_set = CIFAR10(DATASET_ROOT, train=True, transform=transform, download=True) -test_set = CIFAR10(DATASET_ROOT, train=False, transform=transform, download=True) +test_transform = transforms.Compose([ + transforms.ToTensor() +]) + +train_set = CIFAR10(DATASET_ROOT, train=True, transform=train_transform, + download=True) +test_set = CIFAR10(DATASET_ROOT, train=False, transform=test_transform) train_loader = DataLoader(train_set, batch_size=BATCH_SIZE, shuffle=True, num_workers=N_WORKERS) -- cgit v1.2.3