fix dist dataloader eval (#7777)

PaddlePaddle · Jan 4, 2024 · 4437198 · 4437198
1 parent f93e7da
commit 4437198
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,