提交 75ca5bfe 编写于 作者: M Megvii Engine Team

feat(mge): remove GradManager.detach until it is ready

GitOrigin-RevId: 1302b25b2ad625002d5358176ad90c5827698d1c
上级 544b7983
......@@ -100,21 +100,6 @@ class GradManager:
self._record_param(id(p))
return self
def detach(self, params: list):
r"""
Remove specific registered parameters and callback functions.
:param params: registered parameters
"""
if isinstance(params, Tensor):
params = [params]
for idx, param in enumerate(params):
if id(param) in self._param_dict:
self._param_dict.pop(id(param))
self._call_back_dict.pop(id(param))
else:
logger.warning("params with index {} is not attached.".format(idx))
def _register_after_backward_callback(self, callback):
self._after_backward_callback.append(callback)
return self
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册