Skip to content

Commit 81e70a9

Browse files
author
chenyaofo
committed
[Fix] Fix the bug with wrong transformation in cifar.
1 parent bb71353 commit 81e70a9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

codebase/data/cifar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def _cifar(root, image_size, mean, std, batch_size, num_workers, is_vit, dataset
4848
val_transforms = get_vit_val_transforms(mean, std, image_size)
4949
else:
5050
train_transforms = get_train_transforms(mean, std)
51-
val_transforms = get_train_transforms(mean, std)
51+
val_transforms = get_val_transforms(mean, std)
5252

5353
trainset = dataset_builder(root, train=True, transform=train_transforms, download=True)
5454
valset = dataset_builder(root, train=False, transform=val_transforms, download=True)

0 commit comments

Comments
 (0)