Browse Source

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

GitOrigin-RevId: 1302b25b2a
release-1.1
Megvii Engine Team 4 years ago
parent
commit
75ca5bfe65
1 changed files with 0 additions and 15 deletions
  1. +0
    -15
      imperative/python/megengine/autodiff/grad_manager.py

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

@@ -100,21 +100,6 @@ class GradManager:
self._record_param(id(p)) self._record_param(id(p))
return self 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): def _register_after_backward_callback(self, callback):
self._after_backward_callback.append(callback) self._after_backward_callback.append(callback)
return self return self


Loading…
Cancel
Save