提交 544b7983 编写于 作者: M Megvii Engine Team

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

This reverts commit aa9540e09018697110b672f3772473b68305751c.

GitOrigin-RevId: 2d15e503de0728a35028516561ead04d1b90ae06
上级 176268d2
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册