Отправить тензор Pytorch как сообщение в gRPC - PullRequest
Новая
       75

Отправить тензор Pytorch как сообщение в gRPC

0 голосов
/ 06 мая 2020

Я пытаюсь сделать вызов gRP C, чтобы разделить глубокое обучение между слоями в PyTorch. Для этого мне нужно отправить тензор в мое облако для обработки с помощью вызова gRP C. В настоящее время я пытаюсь использовать в качестве сообщения следующее: 

message Tensor {
    google.protobuf.Any out = 1;
}

и в моем коде pyTorch 

from google.protobuf.any_pb2 import Any
...
request = Any()
...
request.Pack(out)
send_grpc_msg(request)

Здесь type(out) равно <class 'torch.Tensor'>. Однако этот код дает следующую ошибку: 

Traceback (most recent call last):
  File "vgg_inference.py", line 95, in <module>
    out = detect_images(Image.open("images/dog.jpg"))
  File "vgg_inference.py", line 88, in detect_images
    request.Pack(out)
  File "/home/rohit/.local/lib/python3.6/site-packages/google/protobuf/internal/well_known_types.py", line 78, in Pack
    self.type_url = '%s%s' % (type_url_prefix, msg.DESCRIPTOR.full_name)
AttributeError: 'Tensor' object has no attribute 'DESCRIPTOR'

Я также не могу найти хороший ресурс для использования ANY в python.

Спасибо

...