Major bug fix: AGC

vballoli · Mar 3, 2021 · 184a4ef · 184a4ef
1 parent 1513cea
commit 184a4ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nfnets/agc.py b/nfnets/agc.py
@@ -41,7 +41,7 @@ def __init__(self, params, optim: optim.Optimizer, clipping: float = 1e-2, eps:
                 if module_name not in names:
                     raise ModuleNotFoundError(
                         "Module name {} not found in the model".format(module_name))
-            parameters = [{"params": module.parameters()} for name,
+            params = [{"params": module.parameters()} for name,
                           module in model.named_modules() if name not in ignore_agc]
 
         super(AGC, self).__init__(params, defaults)