2020年4月18日
pytorch提取中间层的输出
摘要： "参考" 第一种方法：在构建model的时候return对应的层的输出 第2中方法：当模型用Sequential构建时，则让输入依次通过各个模块，抽取出自己需要的输出 阅读全文
posted @ 2020-04-18 16:46 happy_fan 阅读(6620) 评论(0) 推荐(1) 编辑