Skip to content

Commit dc70df7

Browse files
authored
Fix eval hang (#9052)
* fix pipeline eval * fix eval dataloader_num_workers
1 parent 5ef909d commit dc70df7

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
@@ -1500,15 +1500,15 @@ def get_eval_dataloader(self, eval_dataset: Optional[Dataset] = None) -> DataLoa
15001500
eval_dataset,
15011501
batch_sampler=eval_sampler,
15021502
collate_fn=self.data_collator,
1503-
num_workers=self.args.dataloader_num_workers,
1503+
num_workers=0,
15041504
eval=True,
15051505
)
15061506
else:
15071507
return DataLoader(
15081508
eval_dataset,
15091509
batch_sampler=eval_sampler,
15101510
collate_fn=self.data_collator,
1511-
num_workers=self.args.dataloader_num_workers,
1511+
num_workers=0,
15121512
)
15131513

15141514
def get_test_dataloader(self, test_dataset: Dataset) -> DataLoader:

0 commit comments

Comments
 (0)