torch.optim.Optimizer.register_step_pre_hook

Optimizer.register_step_pre_hook(hook)

注册一个 optimizer step 前置钩子，它将在 optimizer step 之前被调用。

它应该具有以下签名

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

optimizer 参数是正在使用的 optimizer 实例。如果 args 和 kwargs 被前置钩子修改，则修改后的值将作为包含 new_args 和 new_kwargs 的元组返回。

参数

hook (Callable) – 要注册的用户定义钩子。

返回

一个句柄，可以通过调用 handle.remove() 来移除添加的钩子

返回类型

torch.utils.hooks.RemovableHandle