From 44426b18b8dfdf77217f4865789b471999265006 Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Sun, 18 Jun 2023 16:19:07 +0800 Subject: [PATCH] train index:auto kmeans when feature shape too large train index:auto kmeans when feature shape too large --- infer-web.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/infer-web.py b/infer-web.py index e13e77e..0524221 100644 --- a/infer-web.py +++ b/infer-web.py @@ -889,8 +889,8 @@ def train_index(exp_dir1, version19): big_npy_idx = np.arange(big_npy.shape[0]) np.random.shuffle(big_npy_idx) big_npy = big_npy[big_npy_idx] - # if(big_npy.shape[0]>2e5): - if(1): + if(big_npy.shape[0]>2e5): + # if(1): infos.append("Trying doing kmeans %s shape to 10k centers."%big_npy.shape[0]) yield "\n".join(infos) try: @@ -1143,8 +1143,8 @@ def train1key( np.random.shuffle(big_npy_idx) big_npy = big_npy[big_npy_idx] - # if(big_npy.shape[0]>2e5): - if(1): + if(big_npy.shape[0]>2e5): + # if(1): info="Trying doing kmeans %s shape to 10k centers."%big_npy.shape[0] print(info) yield get_info_str(info)