Skip to content

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

License

Notifications You must be signed in to change notification settings

KnetML/MAC-Network

Repository files navigation

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •