Code for training Virtual Rodent

Hi! Would you guys mind sharing the code for training the virtual rodent up to the results we see in the YouTube videos from the original paper (like this for gaps: https://www.youtube.com/watch?v=rFelC_YbeLE&ab_channel=Virtualrodent)?

We'd really appreciate a full training recipe with all hyperparameters. We've been trying to train our agent but it gets nowhere near the behavior in the videos.