Browse Source

revert: feat(mge/grad_manager): add `clear_grad` method for GradManager

This reverts commit aa9540e090.

GitOrigin-RevId: 2d15e503de
release-1.1
Megvii Engine Team 4 years ago
parent
commit
544b798374
2 changed files with 2 additions and 9 deletions
  1. +0
    -8
      imperative/python/megengine/autodiff/grad_manager.py
  2. +2
    -1
      imperative/python/test/unit/autodiff/test_grad_manger.py

+ 0
- 8
imperative/python/megengine/autodiff/grad_manager.py View File

@@ -115,14 +115,6 @@ class GradManager:
else:
logger.warning("params with index {} is not attached.".format(idx))

def clear_grad(self):
r"""
For advanced usage: set the grad attribute to None for registered parameters.
It could be more convenient when there is more than one Optimizer.
"""
for param in self._param_dict.values():
param.grad = None

def _register_after_backward_callback(self, callback):
self._after_backward_callback.append(callback)
return self


+ 2
- 1
imperative/python/test/unit/autodiff/test_grad_manger.py View File

@@ -37,7 +37,8 @@ def test_basic():
np.testing.assert_equal(w.grad.numpy(), [[1], [3], [5]])
np.testing.assert_equal(b.grad.numpy(), [1])

gm.clear_grad()
w.grad = None
b.grad = None
with gm:
p = F.matmul(x, w)
y = p + b


Loading…
Cancel
Save