relational network with Gluon
Gluon inplementation of A simple neural module for relational reasoning
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
