comment

huggingface · Feb 5, 2025 · 3a738f9 · 3a738f9
1 parent a38231f
commit 3a738f9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trl/trainer/grpo_trainer.py b/trl/trainer/grpo_trainer.py
@@ -455,6 +455,7 @@ def _set_signature_columns_if_needed(self):
         if self._signature_columns is None:
             self._signature_columns = ["prompt"]
 
+    # We need a custom sampler that samples the same prompt multiple times
     def _get_train_sampler(self) -> Sampler:
         return RepeatRandomSampler(self.train_dataset, self.num_generations)