nemo_rl.data.datasets.response_datasets

Submodules

• nemo_rl.data.datasets.response_datasets.aime24
• nemo_rl.data.datasets.response_datasets.clevr
• nemo_rl.data.datasets.response_datasets.dapo_math
• nemo_rl.data.datasets.response_datasets.deepscaler
• nemo_rl.data.datasets.response_datasets.geometry3k
• nemo_rl.data.datasets.response_datasets.helpsteer3
• nemo_rl.data.datasets.response_datasets.oai_format_dataset
• nemo_rl.data.datasets.response_datasets.oasst
• nemo_rl.data.datasets.response_datasets.openmathinstruct2
• nemo_rl.data.datasets.response_datasets.refcoco
• nemo_rl.data.datasets.response_datasets.response_dataset
• nemo_rl.data.datasets.response_datasets.squad
• nemo_rl.data.datasets.response_datasets.tulu3

Package Contents

Functions

load_response_dataset

Loads response dataset.

Data

DATASET_REGISTRY
__all__

API

nemo_rl.data.datasets.response_datasets.DATASET_REGISTRY

None

nemo_rl.data.datasets.response_datasets.load_response_dataset(data_config: nemo_rl.data.ResponseDatasetConfig)

Loads response dataset.

nemo_rl.data.datasets.response_datasets.__all__

[‘AIME2024Dataset’, ‘CLEVRCoGenTDataset’, ‘DAPOMath17KDataset’, ‘DAPOMathAIME2024Dataset’, ‘DeepScal…