We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 364f50f commit 19f1077Copy full SHA for 19f1077
1 file changed
codebase/data/cifar.py
@@ -55,7 +55,7 @@ def _cifar(root, image_size, mean, std, batch_size, num_workers, is_vit, dataset
55
val_transforms = get_vit_val_transforms(mean, std, image_size)
56
else:
57
train_transforms = get_train_transforms(mean, std)
58
- train_transforms = get_train_transforms(mean, std)
+ val_transforms = get_train_transforms(mean, std)
59
60
trainset = dataset_builder(root, train=True, transform=train_transforms, download=True)
61
valset = dataset_builder(root, train=False, transform=val_transforms, download=True)
0 commit comments