This is the Pytorch Implementation of Compositional Attention Networks for Machine Reasoning
-
Download and extract CLEVR dataset from: https://cs.stanford.edu/people/jcjohns/clevr/, then put it in folder
data/
-
Create folder weights to store snapshot of the model while training.
-
Read the argparse part in
main.py
for more options and runpython main.py
to get the result of default setting.