chore: lint

andreped · Feb 11, 2024 · 7188e30 · 7188e30
1 parent da1c346
commit 7188e30
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gradient_accumulator/accumulators.py b/gradient_accumulator/accumulators.py
@@ -393,9 +393,9 @@ def apply_gradients(
     @tf.function
     def _apply_agc(self, grad: tf.Tensor, var: tf.Variable) -> tf.Tensor:
         """Applies adaptive gradient clipping to the gradient."""
-        return agc.adaptive_clip_grad([var], [grad], clipvalue=self.clipvalue)[
-            0
-        ]
+        return agc.adaptive_clip_grad(
+            [var], [grad], clip_factor=self.clipvalue
+        )[0]
 
     @tf.function
     def _parse_grad(