search

benfred / implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets
https://benfred.github.io/implicit/
MIT License
3.57k stars 612 forks source link

Fix temp memory calculation on GPU #530

Closed benfred closed 2 years ago

benfred commented 2 years ago

In certain cases we could get an OOM on inference on the GPU models. Fix by calculating the temp memory appropriately.