|
@@ -98,13 +98,13 @@ class FedSSLWithPgFedServer(FedSSLServer):
|
|
|
model = self.decompression(codec.unmarshal(uploaded_content.data))
|
|
|
uploaded_models[client.cid] = model
|
|
|
uploaded_weights[client.cid] = uploaded_content.data_size
|
|
|
- self.receive_param(client)
|
|
|
-
|
|
|
+
|
|
|
uploaded_metrics.append(metric.ClientMetric.from_proto(uploaded_content.metric))
|
|
|
+ self.receive_param()
|
|
|
|
|
|
self.set_client_uploads_train(uploaded_models, uploaded_weights, uploaded_metrics)
|
|
|
|
|
|
- def receive_param(self,client):
|
|
|
+ def receive_param(self):
|
|
|
self.uploaded_ids = []
|
|
|
self.uploaded_grads = {}
|
|
|
self.loss_minuses = {}
|