Skip to content

Commit 86dd11d

Browse files
author
kmbae
committed
Shuffle batch
1 parent 565da60 commit 86dd11d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

data_loader.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def get_loader(root, batch_size, scale_size, num_workers=2,
128128
#Dataset(root, scale_size, "B", skip_pix2pix_processing)
129129
a_data_loader = torch.utils.data.DataLoader(dataset=a_data_set,
130130
batch_size=batch_size,
131-
shuffle=False,
131+
shuffle=True,
132132
num_workers=num_workers)
133133
#b_data_loader = torch.utils.data.DataLoader(dataset=b_data_set,
134134
# batch_size=batch_size,

0 commit comments

Comments
 (0)