Fork me on GitHub
2021年11月17日
pytorch hook 钩子
摘要： 简介 hook是钩子，主要作用是不修改主代码，能通过挂载钩子实现额外功能。 pytorch中，主体就是forward和backward，而额外的功能就是对模型的变量进行操作，如“提取”特征图，“提取”非叶子张量的梯度，修改张量梯度等等。hook功能即不必改变网络输入输出的结构，就能方便地获取、改变网 阅读全文
posted @ 2021-11-17 14:53 fnangle 阅读(755) 评论(0) 推荐(0) 编辑