search

subhadarship / kmeans_pytorch

kmeans using PyTorch
https://subhadarship.github.io/kmeans_pytorch
MIT License
479 stars 77 forks source link

SET NUMBER OF ITERATIONS #28

Open TatianaMoreno47 opened 3 years ago

TatianaMoreno47 commented 3 years ago

I want to choose the number of iterations but when i put iter_limit = xxx it give the error that the parameter doesn't exist

subhadarship commented 3 years ago

The latest version is not yet available on pip! Try installing from source https://github.com/subhadarship/kmeans_pytorch#installation