Skip to content

anirudh9119/RIMs

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 

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

About

Code for "Recurrent Independent Mechanisms"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published