Make niter a named parameter.

Nanne · Jan 23, 2021 · 7756777 · 7756777
1 parent e7ee350
commit 7756777
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -269,7 +269,7 @@ def get_clusters(cluster_set):
 
         print('====> Clustering..')
         niter = 100
-        kmeans = faiss.Kmeans(encoder_dim, opt.num_clusters, niter, verbose=False)
+        kmeans = faiss.Kmeans(encoder_dim, opt.num_clusters, niter=niter, verbose=False)
         kmeans.train(dbFeat[...])
 
         print('====> Storing centroids', kmeans.centroids.shape)