Skip to content

Commit 13cdbfe

Browse files
committed
KMeans index extraction actually works and creates smaller index files.
1 parent aef3232 commit 13cdbfe

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

rvc/train/process/extract_index.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,7 @@
4646
np.random.shuffle(big_npy_idx)
4747
big_npy = big_npy[big_npy_idx]
4848

49-
if big_npy.shape[0] > 2e5 and (
50-
index_algorithm == "Auto" or index_algorithm == "KMeans"
51-
):
49+
if big_npy.shape[0] > 2e5 or index_algorithm == "KMeans":
5250
big_npy = (
5351
MiniBatchKMeans(
5452
n_clusters=10000,

0 commit comments

Comments
 (0)