Pretrained model weights

Provides pretrained model weights for a GQN trained on the Shepard-Metzler 5-parts dataset as provided in the mental-rotation.ipynb notebook.