Enabled thread_num option

louiezzang · Dec 13, 2019 · 7d3894e · 7d3894e
1 parent 19a1bba
commit 7d3894e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/run.sh b/run.sh
@@ -22,6 +22,10 @@ echo "nprobe: $NPROBE"
 
 ROOT="$(pwd)/$(dirname "$0")"
 
+echo "Stop $CONTAINER_NAME..."
+./stop.sh "$CONTAINER_NAME"
+
+echo "Start $CONTAINER_NAME..."
 docker run -d --name $CONTAINER_NAME -it \
          -p $PORT:50051 \
          -v $(pwd):/app \
@@ -35,3 +39,5 @@ docker run -d --name $CONTAINER_NAME -it \
            --max_workers "$MAX_WORKERS" \
            --num_threads "$NUM_THREADS" \
            --nprobe $NPROBE
+
+echo "Started!"
diff --git a/server.py b/server.py
@@ -46,7 +46,8 @@ def main(args):
     if num_threads is not None:
         logging.info("num_threads: %d", num_threads)
     server = grpc.server(futures.ThreadPoolExecutor(max_workers=max_workers))
-    servicer = FaissServer(dim, save_path, keys_path, nprobe)
+    #servicer = FaissServer(dim, save_path, keys_path, nprobe)
+    servicer = FaissServer(dim, save_path, keys_path, nprobe, num_threads)
     pb2_grpc.add_ServerServicer_to_server(servicer, server)
     server.add_insecure_port("[::]:50051")
     server.start()