An implementation of Recurrent Independent Mechanisms (Goyal et al. 2019) in PyTorch.
Anirudh Goyal, Alex Lamb, Jordan Hoffmann, Shagun Sodhani, Sergey Levine, Yoshua Bengio, Bernhard Sch{"o}lkopf
It features adding and copying synthetic task from the paper.
./experiment_copying.sh 600 6 4 50 200 for full training & test run of RIMs on the copying task.
./experiment_adding.sh 600 6 4 50 200 0.2 for full training and test run of RIMs on the adding task.
@article{goyal2019recurrent,
title={Recurrent independent mechanisms},
author={Goyal, Anirudh and Lamb, Alex and Hoffmann, Jordan and Sodhani, Shagun and Levine, Sergey and Bengio, Yoshua and Sch{\"o}lkopf, Bernhard},
journal={arXiv preprint arXiv:1909.10893},
year={2019}
}