资源分类
》 资源 》算法》pytorch-nec

pytorch-nec

2019-09-18 | |  144 |   0 |   0

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.

2.

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


上一篇：chainer-visualization

下一篇：Char-RNN

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • vsepp_tensorflow

    Improving Visual-Semantic Embeddings with Hard ...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...