AIME dataset.
Module Contents
API
-
class nemo_rl.data.datasets.eval_datasets.aime.AIMEDataset(
- variant: Literal[2024, 2025] = '2025',
- prompt_file: Optional[str] = None,
- system_prompt_file: Optional[str] = None,
)
Initialization
-
_rekey(data: dict[str, Any])
