All model code and instructions can be found here:
https://github.com/anayebi/mental-sim.
This repo contains pretrained PyTorch models that are optimized to predict the future state of their environment.
As these models are stimulus-computable, they can be ``dropped in'' on any new video for further neural and behavioral comparisons of your design.
Here's a
video recording that explains our work a bit.
