Fixed index version not being written to the index file on train_index() in infer-web.py (#305)
This commit is contained in:
parent
aadf7443c3
commit
c2039b6eca
@ -801,7 +801,7 @@ def train_index(exp_dir1, version19):
|
|||||||
index.add(big_npy[i : i + batch_size_add])
|
index.add(big_npy[i : i + batch_size_add])
|
||||||
faiss.write_index(
|
faiss.write_index(
|
||||||
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(
|
infos.append(
|
||||||
"成功构建索引,added_IVF%s_Flat_nprobe_%s_%s.index"
|
"成功构建索引,added_IVF%s_Flat_nprobe_%s_%s.index"
|
||||||
|
Loading…
Reference in New Issue
Block a user