Skip to content

Commit 81de41a

Browse files
authored
Revert "fix speed ppt eval hang (#9218)" (#9229)
This reverts commit fbe5bc1.
1 parent 57ce8a7 commit 81de41a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddlenlp/trainer/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1623,7 +1623,7 @@ def get_eval_dataloader(self, eval_dataset: Optional[Dataset] = None) -> DataLoa
16231623
eval_dataset,
16241624
batch_sampler=eval_sampler,
16251625
collate_fn=self.data_collator,
1626-
num_workers=0,
1626+
num_workers=self.args.dataloader_num_workers,
16271627
**additional_configs,
16281628
)
16291629

@@ -1670,7 +1670,7 @@ def get_test_dataloader(self, test_dataset: Dataset) -> DataLoader:
16701670
test_dataset,
16711671
batch_size=self.args.per_device_eval_batch_size * self.world_size,
16721672
collate_fn=self.data_collator,
1673-
num_workers=0,
1673+
num_workers=self.args.dataloader_num_workers,
16741674
**additional_config,
16751675
)
16761676
else:

0 commit comments

Comments
 (0)