Updated run script with max_workers and num_threads parameters

louiezzang · Dec 16, 2019 · 9039d53 · 9039d53
1 parent 7d3894e
commit 9039d53
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -27,11 +27,11 @@ $ ./build.sh cia/faiss-server:latest
 
 ## Run the faiss server
 ```
-$ ./run.sh [YOUR_DOCKER_IMAGE_NAME or YOUR_DOCKER_IMAGE_NAME:VERSION] [YOUR_DOCKER_CONTAINER_NAME] [DIM]
+$ ./run.sh [YOUR_DOCKER_IMAGE_NAME or YOUR_DOCKER_IMAGE_NAME:VERSION] [YOUR_DOCKER_CONTAINER_NAME] [DIM] [MAX_WORKERS] [NUM_THREADS]
 ```
 eg. 
 ```
-$ ./run.sh cia/faiss-server:latest faiss-server 200
+$ ./run.sh cia/faiss-server:latest faiss-server 200 1 1
 ```
 
 

diff --git a/run.sh b/run.sh
@@ -2,14 +2,15 @@
 DOCKER_IMAGE=$1
 CONTAINER_NAME=$2
 DIM=$3
+MAX_WORKERS=$4
+NUM_THREADS=$5
 PORT=50051
 #DIM=200
 SAVE_PATH="data/index/faiss_server.index"
 KEYS_PATH=""
 NO_SAVE="false"
 NPROBE=1
-MAX_WORKERS=10
-NUM_THREADS=10
+
 echo "container_name: $CONTAINER_NAME"
 echo "port: $PORT"
 echo "dim: $DIM"