fix proto bug

2025-04-04 02:54:21 +00:00
parent 7530318919
commit 0b26e614f7
1 changed files with 1 additions and 1 deletions
--- a/verl/trainer/ppo/ray_trainer.py
+++ b/verl/trainer/ppo/ray_trainer.py
@@ -410,7 +410,7 @@ class RayPPOTrainer(object):
        self.val_dataloader = DataLoader(dataset=self.val_dataset,
                                         batch_size=self.config.data.val_batch_size,
                                         shuffle=False,
-                                         drop_last=False,
+                                         drop_last=True,
                                         collate_fn=collate_fn)

        print(f'Size of train dataloader: {len(self.train_dataloader)}')