[BugFix] fix dist dataloader eval (#7778)

PaddlePaddle · Jan 4, 2024 · fe3c052 · fe3c052
1 parent 0658258
commit fe3c052
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py
@@ -1293,6 +1293,9 @@ def get_train_dataloader(self):
         )
 
     def _get_eval_sampler(self, eval_dataset: Dataset):
+        if eval_dataset is None or not has_length(eval_dataset):
+            return None
+
         if self.args.world_size <= 1:
             return paddle.io.BatchSampler(
                 eval_dataset,