torch.optim.Optimizer.register_step_pre_hook¶
- Optimizer.register_step_pre_hook(hook)[源代码]¶
注册一个优化器步长预钩子,该钩子将在优化器步长之前被调用。
它应该具有以下签名
hook(optimizer, args, kwargs) -> None or modified args and kwargs
hook(optimizer, args, kwargs) -> (new_args, new_kwargs)
optimizer
参数是正在使用的优化器实例。如果预钩子修改了 args 和 kwargs,则会将转换后的值作为包含 new_args 和 new_kwargs 的元组返回。参数
- hook (Callable) – 要注册的用户定义钩子。
返回值
- 一个句柄,可用于通过调用
handle.remove()
来删除添加的钩子 返回类型