Return to site

Recurrent Models for Visual Attention

September 21, 2015

In this Torch7 post, we outline how to train a recurrent attention model using a reinforcement learning method called REINFORCE. Reproducing the original paper in Torch7, we were able to obtain 0.85% error on the MNIST dataset. Compared to the paper's reported 1.07% that is a significant margin. The post includes some nice visualizations of the learned attention model and detailed explanation of how to use the training and evaluation scripts.

All Posts

Almost done…

We just sent you an email. Please click the link in the email to confirm your subscription!

OKSubscriptions powered by Strikingly