Update optimizers.py

Bluefog-Lib · May 18, 2020 · 8ed5721 · 8ed5721
1 parent 5e09594
commit 8ed5721
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bluefog/torch/optimizers.py b/bluefog/torch/optimizers.py
@@ -422,7 +422,7 @@ def __init__(self, params, named_parameters):
 
     def _register_hooks(self):
         for param_group in self.param_groups:
-            for p in param_group["params"]:  # Is hook function blocking or not?
+            for p in param_group["params"]:
                 if p.requires_grad:
                     p.register_hook(self._make_hook(p))