文件名称:RIMs:“经常性独立机制”的代码
文件大小:556KB
文件格式:ZIP
更新时间:2024-05-29 03:00:29
Python
循环独立机制 在PyTorch中实施(Goyal等人2019)。 , , , , , , 它具有从纸张添加和复制合成任务的功能。 它还具有用于重现atari结果的代码。 例子 ./experiment_copying.sh 600 6 4 50 200用于在复制任务上对RIM进行全面培训和测试。 ./experiment_adding.sh 600 6 4 50 200 0.2用于在添加任务上对RIM进行全面培训和测试。 @article{goyal2019recurrent, title={Recurrent independent mechanisms}, author={Goyal, Anirudh and Lamb, Alex and Hoffmann, Jordan and Sodhani, Shagun and Levine, Sergey