visual-interaction-networks-pytorch

This's an implementation of deepmind Visual Interaction Networks paper using pytorch