Improved explanation for CustomFedAvgM strategy

adap · Nov 23, 2023 · 7f9ec32 · 7f9ec32
1 parent d09a014
commit 7f9ec32
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/baselines/fedavgm/fedavgm/strategy.py b/baselines/fedavgm/fedavgm/strategy.py
@@ -24,7 +24,14 @@
 
 
 class CustomFedAvgM(FedAvg):
-    """Re-implmentation of FedAvgM."""
+    """Re-implmentation of FedAvgM.
+
+    This implementation of FedAvgM diverges from original (Flwr v1.5.0) implementation.
+    Here, the re-implementation introduces the Nesterov Accelerated Gradient (NAG),
+    same as reported in the original FedAvgM paper:
+
+    https://arxiv.org/pdf/1909.06335.pdf
+    """
 
     def __init__(
         self,