From 3e874c6630ebfa77bc82e3cc2f32046dbd259d97 Mon Sep 17 00:00:00 2001 From: Mangio621 Date: Thu, 18 May 2023 02:42:09 +1000 Subject: [PATCH] Fixed index version not being written to the index file on train_index() in infer-web.py --- infer-web.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/infer-web.py b/infer-web.py index df2f9a4..eda5ce2 100644 --- a/infer-web.py +++ b/infer-web.py @@ -801,7 +801,7 @@ def train_index(exp_dir1, version19): index.add(big_npy[i : i + batch_size_add]) faiss.write_index( index, - "%s/added_IVF%s_Flat_nprobe_%s.index" % (exp_dir, n_ivf, index_ivf.nprobe), + "%s/added_IVF%s_Flat_nprobe_%s_%s.index" % (exp_dir, n_ivf, index_ivf.nprobe, version19), ) infos.append( "成功构建索引,added_IVF%s_Flat_nprobe_%s_%s.index"