Awesome
PyTorch implementation of NEC
This repo contains a PyTorch implementation of a Neural Episodic Control agent.
Environment Setup
-
Install conda for Python 2.7.
conda create --name nec --file requirements/conda_requirements.txt
source activate nec
pip install -r requirements/pip_requirements.txt
- Install PyTorch from source at commit eff5b8b.
Usage
python run_nec.py --env=GYM_ENV_ID
where GYM_ENV_ID is the environment ID of the gym environment you which to train on.
Results
A game of Pong as played using the value function learned from a NEC agent
Plot of total reward per episode of Pong vs. episode number