in

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.

Dependencies

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

Datasets

Training

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

GitHub

https://github.com/onlytailei/PyTorch-value-iteration-networks


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