快捷方式

torch.optim.Optimizer.register_step_post_hook

Optimizer.register_step_post_hook(hook)[源代码]

注册一个优化器步骤后挂钩,该挂钩将在优化器步骤之后调用。

它应该具有以下签名

hook(optimizer, args, kwargs) -> None

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

参数

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

返回值

一个句柄,可用于通过调用 handle.remove() 删除添加的挂钩

返回类型

torch.utils.hooks.RemovableHandle

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深度教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源