← all repositories

MishaLaskin/rad

A research implementation of Reinforcement Learning with Augmented Data (RAD), supporting image-based RL agents (SAC, PPO, CURL) on DM-Control and OpenAI Gym.

419 stars Jupyter Notebook ML FrameworksAgents
rad
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides the official implementation of the RAD paper (Laskin et al., 2020), which combines data augmentations with model-free deep reinforcement learning to improve sample efficiency. It includes implementations of SAC, PPO, and CURL agents, with configurable augmentation pipelines (crop, rotate, flip, etc.) for image-based observations. The codebase trains agents on environments from DM-Control and OpenAI Gym using PyTorch, with configurable hyperparameters via command-line arguments.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.