Fork me on GitHub
2018年6月24日
pytorch hook使用
摘要： 由于pytorch会自动舍弃图计算的中间结果，所以想要获取这些数值就需要使用钩子函数。 钩子函数包括Variable的钩子和nn.Module钩子，用法相似。 - 输出： 记录了y的梯度，然后x.data=记录x的梯度 register_forward_hook & register_backwar 阅读全文
posted @ 2018-06-24 20:56 ranjiewen 阅读(1314) 评论(0) 推荐(0)