This repository is an independent implementation of https://arxiv.org/abs/1704.01222 in pytorch.

download bash download.sh, training: python train.py

• scipy
• pytorch==0.3.1 (not tested in other versions)

Shout out to @wassname who increased the scalibility upon my original KDTree implementation!