gRPC channel.close() занимает 10 секунд

Я пытаюсь использовать следующий код на своем рабочем сервере (который я хочу масштабировать до более чем 500 TPS). Я столкнулся со следующей проблемой, когда завалил сервер множеством запросов. По крайней мере, в 1 запросе из 1000 запросов вызов channel.close() занимает 10-10,5 секунд. Я запускаю код на сервере Flask. В настоящее время для каждого запроса я создаю канал и закрываю его. Пожалуйста, помогите мне с этим.

channel = grpc.insecure_channel(serving_address)
stub = prediction_service_pb2_grpc.PredictionServiceStub(channel)
request = predict_pb2.PredictRequest()
request.model_spec.name = model_name
request.model_spec.signature_name = 'serving_default'
request.inputs['model_2_input'].CopyFrom(
        make_tensor_proto_engine(img_array, dtype=1, shape=[1, 224, 224, 3]))
result = stub.Predict(request, 6.0)
channel.close()

person Sathu Hareesh Babu    schedule 31.10.2019    source источник
comment
Привет Хариш, Пара моментов. 1. Можете ли вы попробовать использовать сервер моделей Tensorflow и проверить, сохраняется ли проблема. 2. Есть ли у вас какая-либо конкретная причина для закрытия канала, потому что это не делается в коде в ссылке, github.com/tensorflow/serving/blob/master/tensorflow_serving/.   -  person Tensorflow Support    schedule 04.11.2019
comment
Привет @TensorflowSupport, 1. Да, мы используем контейнер Tf-Serving на нашем рабочем сервере. 2. Я думаю, что для канала на стороне сервера установлен тайм-аут по умолчанию (т. Е. Контейнер обслуживания tf). Я использовал channel.close(), чтобы решить эту проблему.   -  person Sathu Hareesh Babu    schedule 07.11.2019
comment
@TensorflowSupport также не могли бы вы рассказать мне, как добавить журналы запросов на сервер моделей Tensorflow (контейнер tensorflow_serving)   -  person Sathu Hareesh Babu    schedule 07.11.2019
comment
Вы можете использовать --v=1 для журналов. См. github.com/tensorflow/serving/issues/285 и stackoverflow.com/questions/46359852/ для получения дополнительной информации. Да, журналы могут помочь нам понять проблему.   -  person Tensorflow Support    schedule 08.11.2019