search

DeMoriarty / TorchPQ

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
MIT License
214 stars 21 forks source link

Fix undefined device variable #2

Closed francisr closed 3 years ago

francisr commented 3 years ago

device -> data.device.