Fixing used val_transform in barlow_twins.py (#367)

karellat · web-flow · commit a7d947178455 · 2024-12-03T19:34:32.000Z
diff --git a/lightning_examples/barlow-twins/barlow_twins.py b/lightning_examples/barlow-twins/barlow_twins.py
@@ -118,7 +118,7 @@ def cifar10_normalization():
 val_transform = BarlowTwinsTransform(
     train=False, input_height=32, gaussian_blur=False, jitter_strength=0.5, normalize=cifar10_normalization()
 )
-val_dataset = CIFAR10(root=".", train=False, download=True, transform=train_transform)
+val_dataset = CIFAR10(root=".", train=False, download=True, transform=val_transform)
 
 train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, num_workers=num_workers, drop_last=True)
 val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False, num_workers=num_workers, drop_last=True)

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ def cifar10_normalization():`
`118`	`118`	`val_transform = BarlowTwinsTransform(`
`119`	`119`	`train=False, input_height=32, gaussian_blur=False, jitter_strength=0.5, normalize=cifar10_normalization()`
`120`	`120`	`)`
`121`		`-val_dataset = CIFAR10(root=".", train=False, download=True, transform=train_transform)`
	`121`	`+val_dataset = CIFAR10(root=".", train=False, download=True, transform=val_transform)`
`122`	`122`
`123`	`123`	`train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, num_workers=num_workers, drop_last=True)`
`124`	`124`	`val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False, num_workers=num_workers, drop_last=True)`