Format code (#221)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2023-05-05 13:13:41 +08:00
committed by GitHub
parent ccf6e6bbd2
commit 6726af00cf
2 changed files with 11 additions and 11 deletions

View File

@@ -694,9 +694,9 @@ def train_index(exp_dir1):
# faiss.write_index(index, '%s/trained_IVF%s_Flat_FastScan.index'%(exp_dir,n_ivf))
infos.append("adding")
yield "\n".join(infos)
batch_size_add=8192
for i in range(0,big_npy.shape[0],batch_size_add):
index.add(big_npy[i:i+batch_size_add])
batch_size_add = 8192
for i in range(0, big_npy.shape[0], batch_size_add):
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),
@@ -915,9 +915,9 @@ def train1key(
"%s/trained_IVF%s_Flat_nprobe_%s.index" % (exp_dir, n_ivf, index_ivf.nprobe),
)
yield get_info_str("adding index")
batch_size_add=8192
for i in range(0,big_npy.shape[0],batch_size_add):
index.add(big_npy[i:i+batch_size_add])
batch_size_add = 8192
for i in range(0, big_npy.shape[0], batch_size_add):
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),
@@ -1544,4 +1544,4 @@ with gr.Blocks() as app:
inbrowser=not config.noautoopen,
server_port=config.listen_port,
quiet=True,
)
)