Skip to content

Gluon implement of "A simple neural network module of relational reasoning"

Notifications You must be signed in to change notification settings

seujung/relational-network-gluon

Repository files navigation

relational network with Gluon


Gluon inplementation of A simple neural module for relational reasoning

net_structure

Requirements

  • Python 3.6.1
  • Mxnet(1.2)
  • tqdm
  • opencv-python

Usage

  • arguments
    • batch_size : Define batch size (defualt=64)
    • epoches : Define total epoches (default=50)
    • GPU_COUNT : Use GPU count (default=2)
    • show_status : show loss and accuracy for each epoch (default=True)
data generation
python sort-of-clevr.py
default setting
python main.py

or

manual setting
python main.py --batch_size=32 --epoches=100

Results

perf_acc

perf_loss

Reference

About

Gluon implement of "A simple neural network module of relational reasoning"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published