nemo_rl.data.datasets.response_datasets.deepscaler

Module Contents

Classes

DeepScalerDataset

Simple wrapper around the DeepScaler dataset with train split.

API

class nemo_rl.data.datasets.response_datasets.deepscaler.DeepScalerDataset(**kwargs)

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the DeepScaler dataset with train split.

Initialization

format_data(data: dict[str, Any]) → dict[str, Any]