Fixed index version not being written to the index file on train_index() in infer-web.py (#305)

2023-05-18 20:02:12 +10:00 · 2023-05-18 20:02:12 +10:00 · c2039b6eca
parent aadf7443c3
commit c2039b6eca
1 changed files with 1 additions and 1 deletions
--- 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"