Skip to content

Commit 19f1077

Browse files
committed
fix bugs
1 parent 364f50f commit 19f1077

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
@@ -55,7 +55,7 @@ def _cifar(root, image_size, mean, std, batch_size, num_workers, is_vit, dataset
5555
val_transforms = get_vit_val_transforms(mean, std, image_size)
5656
else:
5757
train_transforms = get_train_transforms(mean, std)
58-
train_transforms = get_train_transforms(mean, std)
58+
val_transforms = get_train_transforms(mean, std)
5959

6060
trainset = dataset_builder(root, train=True, transform=train_transforms, download=True)
6161
valset = dataset_builder(root, train=False, transform=val_transforms, download=True)

0 commit comments

Comments
 (0)