To appear in NeurIPS 2019.
Abstract Reasoning with Distracting Features
(Kecheng Zheng, Wei Wei and Zheng-jun Zha, 2019)
To download the dataset, please check chizhang's project page.
For details, please check our paper.
Important
- Python 2.7
- PyTorch
- CUDA and cuDNN
python main.py --net <model name> --datapath <path to the dataset> --rl False --typeloss False
#TODO
- The code for PGM dataset
- The long-term features (state) in teacher model
If you find the paper and/or the code helpful, please cite us.
@inproceedings{zheng2019abstract,
title={Abstract Reasoning with Distracting Features},
author={Kecheng Zheng and Zheng-jun Zha and Wei Wei},
booktitle={Advances in Neural Information Processing Systems},
year={2019}
}
Please feel free to discuss paper/code through issues or emails.