钩子函数

来源:3-32 Pytorch与autograd中的几个重要概念-autograd例子

Amberzy

2020-07-29

x = register_hook(lambda grad:grad*2)

这个钩子函数是在计算梯度的时候作用的吗?

写回答

1回答

会写代码的好厨师

2020-08-15

这个操作主要是执行lamda这个函数,也即是梯度*2的时候,会被执行到。

0
1
Amberzy
非常感谢!
2020-08-21
共1条回复

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程