torch.optim.Optimizer.register_load_state_dict_post_hook#

Optimizer.register_load_state_dict_post_hook(hook, prepend=False)[source]#

注册一个 load_state_dict 后续钩子，该钩子将在调用 load_state_dict() 后被调用。它应该具有以下签名

hook(optimizer) -> None

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

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

参数

hook (Callable) – 用户定义的待注册钩子。
prepend (bool) – 如果为 True，则提供的 load_state_dict 的后续钩子将比所有已注册的后续钩子先执行。否则，提供的 hook 将在所有已注册的后续钩子之后执行。(默认: False)

返回

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

返回类型

torch.utils.hooks.RemoveableHandle

文档