Running RIMs on Atari

[aayushee]

Hello

I am trying to run the blocks_atari code but cannot do so. On using environments like zaxxon no frame skip v4 and beamrider no frame skip v4, I get a matrix multiplication error.

And for other environments like Walker v2, Reacher v2, Hopper v2, HalfCheetah v2, there is an assertion error because number of hidden inputs is not completely divisible by number of topk RIMs.

Could you please share how this issue can be resolved?