mjacar / pytorch-nec

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

License:MIT License


Languages

Language:Python 100.0%