1
0
mirror of synced 2024-11-28 01:10:56 +01:00

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

This commit is contained in:
Cole Mangio 2023-05-18 20:02:12 +10:00 committed by GitHub
parent aadf7443c3
commit c2039b6eca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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"