oumi.datasets.grpo

GRPO datasets module.

class oumi.datasets.grpo.TldrGrpoDataset(*, dataset_name: str | None = None, dataset_path: str | None = None, split: str | None = None, tokenizer: PreTrainedTokenizerBase | None = None, return_tensors: bool = False, **kwargs)

Bases: BaseExperimentalGrpoDataset

Dataset class for the trl-lib/tldr dataset.

dataset_name: str

default_dataset: str | None = 'trl-lib/tldr'

trust_remote_code: bool

Subpackages

• oumi.datasets.grpo.rewards
  • compute_sharp_target_token_length_reward()
  • compute_soft_target_token_length_reward()