Skip to content

mjacar/pytorch-nec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch implementation of NEC

This repo contains a PyTorch implementation of a Neural Episodic Control agent.

Environment Setup

  1. 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
  1. 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

nec_pong_gif

A game of Pong as played using the value function learned from a NEC agent

nec_pong_png

Plot of total reward per episode of Pong vs. episode number

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages