trained on 12 different geometric structures 1200 steps geopsyc https://imgur.com/mjoNf6E