Merge pull request #5477 from FederatedAI/feature-2.0.0-rc-sshe-glm

fix lr loss(#5466)
FederatedAI · Jan 31, 2024 · 5830689 · 5830689
2 parents f412237 + 0ee456a
commit 5830689
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fate/ml/glm/hetero/coordinated_lr/guest.py b/python/fate/ml/glm/hetero/coordinated_lr/guest.py
@@ -290,7 +290,7 @@ def asynchronous_compute_gradient(self, batch_ctx, encryptor, w, X, Y, weight):
         loss += torch.matmul((1 / h * Xw).T, Xw_h) - torch.matmul((2 / h * Y).T, Xw_h)
 
         for Xw2_h in batch_ctx.hosts.get("Xw2_h"):
-            loss += 0.125 / h * Xw2_h
+            loss += 2 / h * Xw2_h
         h_loss_list = batch_ctx.hosts.get("h_loss")
         for h_loss in h_loss_list:
             if h_loss is not None: