nemo_rl.data.datasets.response_datasets

Submodules

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

Package Contents

Functions

load_response_dataset

Loads response dataset.

Data

__all__

API

nemo_rl.data.datasets.response_datasets.load_response_dataset(data_config, seed: int = 42)

Loads response dataset.

nemo_rl.data.datasets.response_datasets.__all__

[‘CLEVRCoGenTDataset’, ‘DeepScalerDataset’, ‘DAPOMath17KDataset’, ‘Geometry3KDataset’, ‘OpenAIFormat…