/gem

Primary LanguagePython

Generalizable Episodic Memory For Deep Reinforcement Learning

This is the anonymous github containing code of ICML 2021 Submission Paper (ID 2335): "Generalizable Episodic Memory For Deep Reinforcement Learning"

Installation

GEM uses 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 that these scripts assume you download this git into your home directory.