Datasets

TorchRL provides dataset utilities for offline RL and data management.

datasets.AtariDQNExperienceReplay(dataset_id)	Atari DQN Experience replay class.
datasets.D4RLExperienceReplay(dataset_id, ...)	An Experience replay class for D4RL.
datasets.GenDGRLExperienceReplay(dataset_id)	Gen-DGRL Experience Replay dataset.
datasets.MinariExperienceReplay(dataset_id, ...)	Minari Experience replay dataset.
datasets.OpenMLExperienceReplay(name, batch_size)	An experience replay for OpenML data.
datasets.OpenXExperienceReplay(dataset_id[, ...])	Open X-Embodiment datasets experience replay.
datasets.RobosetExperienceReplay(dataset_id, ...)	Roboset experience replay dataset.
datasets.VD4RLExperienceReplay(dataset_id, ...)	V-D4RL experience replay dataset.