This repository includes a batch computation extension implementation of GPU-SME-kNN and supplies a pytorch interface. Note that all of the code can be only used for research purposes.
CUDA>=9.0
pytorch
This repository includes a batch computation extension implementation of GPU-SME-kNN and supplies a pytorch interface. Note that all of the code can be only used for research purposes.
CUDA>=9.0
pytorch