anirudh9119 / RIMs

Code for "Recurrent Independent Mechanisms"

Recurrent Independent Mechanisms

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. It also features the code to reproduce atari results.

Examples

./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}
}