fix: 修复一些已知问题

2026-03-30 01:08:19 +08:00
parent c01791ff54
commit 60f4df95bd
4 changed files with 54 additions and 32 deletions
@@ -171,13 +171,13 @@ class GRPODataset(BaseDataset):
    def __getitem__(self, index: int) -> Dict[str, Tensor]:
        begin_idx, end_idx = self.get_index(index)

-        prompts =  self._fetch_data(begin_idx, end_idx, "prompts"),
-        responses = self._fetch_data(begin_idx, end_idx, "responses"),
-        masks = self._fetch_data(begin_idx, end_idx, "masks"),
-        rewards =  self._fetch_data(begin_idx, end_idx, "rewards")
-        
-        return {"prompts": prompts, "responses": responses, "masks": masks, "rewards": rewards}
+        prompts = self._fetch_data(begin_idx, end_idx, "prompts")
+        responses = self._fetch_data(begin_idx, end_idx, "responses")
+        masks = self._fetch_data(begin_idx, end_idx, "masks")
+        rewards = self._fetch_data(begin_idx, end_idx, "rewards")
    
+        return {"prompts": prompts, "responses": responses, "masks": masks, "rewards": rewards}
+

 class DatasetLoader:
    @staticmethod