From 75ca5bfe65c78085d047d1a5c516a0bba3f8b7e6 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 17 Nov 2020 16:25:49 +0800 Subject: [PATCH] feat(mge): remove GradManager.detach until it is ready GitOrigin-RevId: 1302b25b2ad625002d5358176ad90c5827698d1c --- .../python/megengine/autodiff/grad_manager.py | 15 --------------- 1 file changed, 15 deletions(-) diff --git a/imperative/python/megengine/autodiff/grad_manager.py b/imperative/python/megengine/autodiff/grad_manager.py index 0cd44b38..db46b12e 100644 --- a/imperative/python/megengine/autodiff/grad_manager.py +++ b/imperative/python/megengine/autodiff/grad_manager.py @@ -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 -- GitLab