This repo contains a pytorch implementation of the supervised half of Banino et al's 2018 Nature paper: Vector-based navigation using grid-like representations in artificial agents.
https://www.nature.com/articles/s41586-018-0102-6
Original implementation and DATA can be found here: https://github.com/deepmind/grid-cells
run convert_data.py to convert tensorflow data to pytorch format