torch.optim.Optimizer.register_load_state_dict_post_hook

Optimizer.register_load_state_dict_post_hook(hook, prepend=False)[源代码]

注册一个在调用load_state_dict()之后触发的 post-hook。它应具有以下签名：

hook(optimizer) -> None

optimizer 参数是指正在使用的优化器实例。

在调用 self.load_state_dict() 之后，该钩子将被调用，并使用参数 self。注册的钩子可以在 load_state_dict 加载了 state_dict 之后执行后处理。

参数

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

• prepend (bool) – 如果为 True，则提供的后置钩子 hook 将在调用 load_state_dict 时，在所有已注册的后置钩子之前触发。否则，该钩子将在所有已注册的后置钩子之后触发。（默认值：False）

返回值

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

返回类型

torch.utils.hooks.RemoveableHandle