Generalizable Episodic Memory For Deep Reinforcement Learning

This is the github containing code of ICML 2021 Paper: "Generalizable Episodic Memory For Deep Reinforcement Learning"

Installation

We use python 3.7 with requirements listed in requirements.txt

we recommend using a conda virtual environment to install packages

Running

To run MuJoCO , please refer to ./gem_mujoco/run/run.sh

To run Atari , please refer to /gem_atari/run_gem.sh

Note these scripts assume that you download this git into your home directory.