Skip to content

Commit 05c779a

Browse files
author
Azure Pipelines
committed
Merge remote-tracking branch 'origin/main' into publication
2 parents 639e4ad + a7d9471 commit 05c779a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightning_examples/barlow-twins/barlow_twins.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def cifar10_normalization():
118118
val_transform = BarlowTwinsTransform(
119119
train=False, input_height=32, gaussian_blur=False, jitter_strength=0.5, normalize=cifar10_normalization()
120120
)
121-
val_dataset = CIFAR10(root=".", train=False, download=True, transform=train_transform)
121+
val_dataset = CIFAR10(root=".", train=False, download=True, transform=val_transform)
122122

123123
train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, num_workers=num_workers, drop_last=True)
124124
val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False, num_workers=num_workers, drop_last=True)

0 commit comments

Comments
 (0)