/relation-networks-tensorflow

WIP) A TensorFlow implementation of DeepMind's Relation Networks (https://arxiv.org/abs/1706.01427)

Primary LanguagePython

Relation Networks in Tensorflow (WIP)

A TensorFlow implementation of DeepMind's Relation Networks.

THIS PROJECT IS A WORK IN PROGRESS : Not working yet!

architecture

Prerequisites

Downloading the dataset

You need to download CLEVR dataset.
The main dataset is 18GB - with traning set which has 70,000 images and about 700,000 questions.

Training

./train.py

License: MIT