Pytorch github. 2 Training steps are much similar to rpautrat/Superpoint.