Shortcuts

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources