Value Iteration Networks in PyTorch

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.


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



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

python --lr 0.001



Leave a Reply

Your email address will not be published. Required fields are marked *

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.

GIPHY App Key not set. Please check settings

How Medical Companies are Innovating Through Agile Practices

Vodafone Idea to Get Funding Soon: Takkar