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])
|
||||
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"
|
||||
|
Loading…
Reference in New Issue
Block a user