Client pull weights from server. Namely server push weights from clients.
(self)
| 99 | self.sock.close() |
| 100 | |
| 101 | def pull(self) -> None: |
| 102 | """Client pull weights from server. |
| 103 | |
| 104 | Namely server push weights from clients. |
| 105 | """ |
| 106 | message = proto.WeightsExchange() |
| 107 | message = utils.receive_message(self.sock, message) |
| 108 | for k, v in message.weights.items(): |
| 109 | self.weights[k] = utils.deserialize_tensor(v) |
| 110 | |
| 111 | def push(self) -> None: |
| 112 | """Client push weights to server. |