.

BaguaSys · Dec 5, 2022 · 343516e · 343516e
1 parent a65aace
commit 343516e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/bagua/torch_api/algorithms/q_adam.py b/bagua/torch_api/algorithms/q_adam.py
@@ -65,7 +65,6 @@ def step(self, closure=None):
         for group_id, group in enumerate(self.param_groups):
 
             lr = group["lr"]
-            weight_decay = group["weight_decay"]
             beta1, beta2 = group["betas"]
             eps = group["eps"]