KnetML / MAC-Network

Memory, Attention and Composition (MAC) Network for CLEVR implemented via KnetLayers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Compositional Attention Networks for Machine Reasoning

Knet implementation of the paper "Compositional attention networks for machine reasoning." Hudson, Drew A., and Christopher D. Manning.

Running Demo

Open visualize.ipynb notebook with jupyter to see the demo.

Getting Data

You have two options for the data setup.

a) Raw Data

1-Download CLEVR dataset to data/ folder.

2-Process the CLEVR data:

julia trainsetup.jl data/CLEVR_v1.0

b) Processed Data

1-Download preprocessed data:

julia trainsetup.jl

Training

Below configuration can achieve %98.27 accuracy on CLEVR dataset

julia train.jl src/main.jl configs/config2.jl

About

Memory, Attention and Composition (MAC) Network for CLEVR implemented via KnetLayers

License:MIT License


Languages

Language:Julia 86.3%Language:Jupyter Notebook 12.8%Language:Shell 0.9%