Skip to content

Commit 1a57fb4

Browse files
committed
update
1 parent c5b36d5 commit 1a57fb4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/loops/test_double_iter_in_iterable_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def create_queue() -> Queue:
4848

4949

5050
def train_model(queue: Queue, max_epochs: int, ckpt_path: Path) -> None:
51-
dataloader = DataLoader(QueueDataset(queue), num_workers=1, batch_size=None, persistent_workers=True)
51+
dataloader = DataLoader(QueueDataset(queue), num_workers=1, batch_size=None)
5252
trainer = Trainer(
5353
max_epochs=max_epochs,
5454
enable_progress_bar=False,

0 commit comments

Comments
 (0)