adaptive n_kmeans for MultiKmeans

DeMoriarty / fast_pytorch_kmeans

This is a pytorch implementation of k-means clustering algorithm

MIT License

284 stars 38 forks source link

adaptive n_kmeans for MultiKmeans #3

Closed hardyho closed 1 year ago

hardyho commented 2 years ago

Hi! Would it be possible to deprecate this n_kmeans as a param and instead get this param during forward and adapt it to the shape of the input tensor? Thanks!

DeMoriarty commented 1 year ago

fixed