From 71427575c4f419458df3460a526becd9cafe5fd9 Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Tue, 2 May 2023 12:17:09 +0000 Subject: [PATCH] Update infer-web.py --- infer-web.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/infer-web.py b/infer-web.py index 394a4c9..4ac2f3d 100644 --- a/infer-web.py +++ b/infer-web.py @@ -670,7 +670,7 @@ def train_index(exp_dir1): phone = np.load("%s/%s" % (feature_dir, name)) npys.append(phone) big_npy = np.concatenate(npys, 0) - # np.save("%s/total_fea.npy" % exp_dir, big_npy) + np.save("%s/total_fea.npy" % exp_dir, big_npy) # n_ivf = big_npy.shape[0] // 39 n_ivf = min(int(16 * np.sqrt(big_npy.shape[0])), big_npy.shape[0] // 39) infos = [] @@ -892,7 +892,7 @@ def train1key( phone = np.load("%s/%s" % (feature_dir, name)) npys.append(phone) big_npy = np.concatenate(npys, 0) - # np.save("%s/total_fea.npy" % exp_dir, big_npy) + np.save("%s/total_fea.npy" % exp_dir, big_npy) # n_ivf = big_npy.shape[0] // 39 n_ivf = min(int(16 * np.sqrt(big_npy.shape[0])), big_npy.shape[0] // 39) yield get_info_str("%s,%s" % (big_npy.shape, n_ivf))