torch.optim.Optimizer.register_step_pre_hook

Optimizer.register_step_pre_hook(hook)[源代码]

注册一个在优化器步骤执行前被调用的预处理钩子。

它应具有如下签名：

hook(optimizer, args, kwargs) -> None or modified args and kwargs

optimizer 参数是指正在使用的优化器实例。如果预处理钩子修改了 args 和 kwargs，则会返回一个新的包含修改后参数的元组，即新的 new_args 和 new_kwargs。