tangger/Search-R1
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix proto bug

Browse Source
This commit is contained in:
PeterGriffinJin
2025-04-04 02:54:21 +00:00
parent 7530318919
commit 0b26e614f7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
verl/trainer/ppo/ray_trainer.py
View File

@@ -410,7 +410,7 @@ class RayPPOTrainer(object):
self.val_dataloader = DataLoader(dataset=self.val_dataset, self.val_dataloader = DataLoader(dataset=self.val_dataset,
batch_size=self.config.data.val_batch_size, batch_size=self.config.data.val_batch_size,
shuffle=False, shuffle=False,
drop_last=False, drop_last=True,
collate_fn=collate_fn) collate_fn=collate_fn)
print(f'Size of train dataloader: {len(self.train_dataloader)}') print(f'Size of train dataloader: {len(self.train_dataloader)}')