rbunn80110 / arc-pytorch

PyTorch implementation of Attentive Recurrent Comparators

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

arc-pytorch

PyTorch implementation of Attentive Recurrent Comparators by Shyam et al.

A blog explaining Attentive Recurrent Comparators

Visualizing Attention

On Same characters

On Different Characters

How to run?

Download data

python download_data.py

A one-time 52MB download. Shouldn't take more than a few minutes.

Train

python train.py --cuda

Let it train until the accuracy rises to at least 80%

Visualize

python viz.py --cuda --load 0.13591022789478302 --same

Run with exactly the same parameters as train.py and specify the model to load. Specify "--same" if you want to generate a sample with same characters in both images.

About

PyTorch implementation of Attentive Recurrent Comparators

License:MIT License


Languages

Language:Python 100.0%