Skip to content

Commit 83bcf13

Browse files
committed
fix dist dataloader eval
1 parent f93e7da commit 83bcf13

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddlenlp/trainer/trainer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1293,6 +1293,9 @@ def get_train_dataloader(self):
12931293
)
12941294

12951295
def _get_eval_sampler(self, eval_dataset: Dataset):
1296+
if eval_dataset is None or not has_length(eval_dataset):
1297+
return None
1298+
12961299
if self.args.world_size <= 1:
12971300
return paddle.io.BatchSampler(
12981301
eval_dataset,

0 commit comments

Comments
 (0)