def send_proto(self, iter_id, name, proto):
any_proto = google.protobuf.any_pb2.Any()
any_proto.Pack(proto)
msg = tws_pb.TrainerWorkerMessage(data=tws_pb.DataMessage(
iter_id=iter_id, name=name, any_data=any_proto))
self._transmit(msg)
logging.debug('Data: send protobuf %s for iter %d. seq_num=%d.', name,
iter_id, msg.seq_num)
def send(self, iter_id, name, x):
msg = tws_pb.TrainerWorkerMessage(data=tws_pb.DataMessage(
iter_id=iter_id, name=name, tensor=tf.make_tensor_proto(x)))
self._transmit(msg)
logging.debug('Data: send %s for iter %d. seq_num=%d.', name, iter_id,
msg.seq_num)
