一个训练好的、庞大的网络模型,我想提取它第一个卷积层的输出,得怎么做?
目前查阅了文档和论坛之前的文章也没找到解决方案
找到解决方法了,已知网络结构子层的名字情况下,使用register_forward_post_hook打印输出
学习啦
学习啦,论文做可视化会用到,之前还头疼过
找到解决方法了,已知网络结构子层的名字情况下,使用register_forward_post_hook打印输出
学习啦
学习啦,论文做可视化会用到,之前还头疼过