无颜祖

pytorch

denoise(1) linux(2) qrestore项目总结记录(1)

/ 注册

全部文章 / pytorch （共1篇）

使用pytorch hook可视化cnn

利用pytorch中的钩子（hook），我们不用改变输入输出中间的网络结构，可以方便的获取、改变网络中间层变量的值和梯度。本篇文章记录了如何利用register_forward_hook(hook)获取pre-trained模型任意层的输出特征图并保存。 register_forward_hook(...

2019-08-04

1 2701