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
参数是正在使用的优化器实例。在对
self
调用load_state_dict
后,将使用参数self
调用该钩子。注册的钩子可用于在load_state_dict
加载state_dict
后执行后处理。- 参数
hook (Callable) – 要注册的用户定义的钩子。
prepend (bool) – 如果为 True,则提供的后
hook
将在load_state_dict
上所有已注册的后挂钩之前触发。否则,提供的hook
将在所有已注册的后挂钩之后触发。(默认值:False)
- 返回值
一个句柄,可用于通过调用
handle.remove()
删除添加的钩子- 返回类型
torch.utils.hooks.RemoveableHandle