torch.nn.modules.module.register_module_backward_hook
-
torch.nn.modules.module.register_module_backward_hook(hook)[source][source]
向所有模块注册通用的反向传播钩子。
此功能已被弃用,推荐使用 torch.nn.modules.module.register_module_full_backward_hook()
,此函数的行为将在未来版本中发生变化。
- 返回值:
一个可以用来通过调用 handle.remove()
移除已添加钩子的句柄。
- 返回类型:
torch.utils.hooks.RemovableHandle