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

Mangio621 · web-flow · commit c2039b6eca9d · 2023-05-18T10:02:12.000Z
diff --git 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"

Original file line number	Diff line number	Diff line change
`@@ -801,7 +801,7 @@ def train_index(exp_dir1, version19):`
`801`	`801`	`index.add(big_npy[i : i + batch_size_add])`
`802`	`802`	`faiss.write_index(`
`803`	`803`	`index,`
`804`		`- "%s/added_IVF%s_Flat_nprobe_%s.index" % (exp_dir, n_ivf, index_ivf.nprobe),`
	`804`	`+ "%s/added_IVF%s_Flat_nprobe_%s_%s.index" % (exp_dir, n_ivf, index_ivf.nprobe, version19),`
`805`	`805`	`)`
`806`	`806`	`infos.append(`
`807`	`807`	`"成功构建索引，added_IVF%s_Flat_nprobe_%s_%s.index"`