Skip to content

Commit 672fd6d

Browse files
committed
add this file by mistake
1 parent 7974f9d commit 672fd6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gallery/transforms/plot_cutmix_mixup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def collate_fn(batch):
112112
return cutmix_or_mixup(*default_collate(batch))
113113

114114

115-
dataloader = DataLoader(dataset, batch_size=4, shuffle=True, num_workers=0, collate_fn=collate_fn)
115+
dataloader = DataLoader(dataset, batch_size=4, shuffle=True, num_workers=2, collate_fn=collate_fn)
116116

117117
for images, labels in dataloader:
118118
print(f"{images.shape = }, {labels.shape = }")

0 commit comments

Comments
 (0)