Skip to content

Commit 9582481

Browse files
committed
add batchsize
1 parent 8cb363a commit 9582481

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

README_ENVECTOR.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ python -m vectordb_bench.cli.vectordbbench envectorflat \
125125
--db-label "Performance1536D500K-FLAT"
126126

127127
# ivf: IVF-FLAT with random centroids
128+
export NUM_PER_BATCH=500000 # set to the database size for efficiency
128129
python -m vectordb_bench.cli.vectordbbench envectorivfflat \
129130
--uri "localhost:50050" \
130131
--case-type "Performance1536D500K" \

scripts/run_benchmark.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ if [[ -z "$REQUESTED_TYPE" || "$REQUESTED_TYPE" == "flat" ]]; then
6969
fi
7070

7171
if [[ -z "$REQUESTED_TYPE" || "$REQUESTED_TYPE" == "ivf" ]]; then
72+
export NUM_PER_BATCH=500000 # set database size for efficiency
7273
run_case envectorivfflat "$DATASET_NAME-IVF-RANDOM" \
7374
--nlist "$NLIST" \
7475
--nprobe "$NPROBE"

0 commit comments

Comments
 (0)