Skip to content

PyTorch implementation of the Value Iteration Networks (VIN) (NIPS '16 best paper)

Notifications You must be signed in to change notification settings

onlytailei/Value-Iteration-Networks-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tamar, A., Wu, Y., Thomas, G., Levine, S., and Abbeel, P. Value Iteration Networks. Neural Information Processing Systems (NIPS) 2016

This repository contains an implementation of Value Iteration Networks (VIN) in PyTorch based on the original Theano implementation by the authors and the TensoFlow implementation by Abhishek Kumar.

VIN won the Best Paper Award at NIPS 2016.

Value Iteration Network and Module

Dependencies

  • Python 2.7
  • PyTorch
  • SciPy >= 0.18.1 (to load the data)

Datasets

Training

python train_main.py

Several arguments can be set in train_main.py like learning rate. Please check train_main.py for details.

python train_main.py --lr 0.001

References

About

PyTorch implementation of the Value Iteration Networks (VIN) (NIPS '16 best paper)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages