Skip to content

Commit fe3c052

Browse files
authored
[BugFix] fix dist dataloader eval (#7778)
1 parent 0658258 commit fe3c052

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)