Skip to content

Commit 0bdcb63

Browse files
committed
remove parameter "shuffle" from dataloader_params because it would collide with fixed shuffle=False
1 parent cd3b4a9 commit 0bdcb63

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pytorch_ie/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ def _sanitize_parameters(
195195
forward_parameters[p_name] = pipeline_parameters[p_name]
196196

197197
# set dataloader parameters
198-
for p_name in ["batch_size", "num_workers", "shuffle"]:
198+
for p_name in ["batch_size", "num_workers"]:
199199
if p_name in pipeline_parameters:
200200
dataloader_params[p_name] = pipeline_parameters[p_name]
201201

0 commit comments

Comments
 (0)