输出图片变化的代码提问

[NwTbbetter]

如果跑大型的模型，想要输出图片经过每个卷积层的变化，应该把输出日志的代码写在哪里呀

是ppdet/engine/trainer.py里的def trian（）嘛

具体的实现代码应该怎么写呀

[rainyfly]

你好，如果想要采集这些数据的话，或许会比较麻烦，最简单的思路就是在模型（nn.Layer）实现的forward方法里，在每个卷积层的输出后面加入记录的代码。或许你可以自己写个post hook, 这个post hook用来记录每个Layer它forward后的输出，关于post hook的作用和怎么写可以参考 https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/nn/Layer_cn.html#register-forward-post-hook-hook