Non-metric 3D visual and spatial representations Object centric representation in 3D Make sure that args.r is a multiple of batch size. Batch size should also be a factor of dataset length (train)