search

DeMoriarty / fast_pytorch_kmeans

This is a pytorch implementation of k-means clustering algorithm
MIT License
279 stars 35 forks source link

Refactored work splitting in `max_sim`, removed pynvml dependency #22

Closed DeMoriarty closed 1 year ago