获取网络中间几层的结果IntermediateLayerGetter()及源码分析
获取网络中间几层的结果IntermediateLayerGetter()及源码分析
思想:先创建一个model ,然后把它传入IntermediateLayerGetter中,并传入一个字典,传入字典的key是model的直接的层,传入字典的value是返回字典中的key,返回字典的value对应的是model运行的中间结果。
一个小技巧是,传入的字典期望是str–str,如果传入str–int,那么使用的方式和字典是一样的。
注意:因为 model.named_children() 只能找到直接下一层的名
原创
2021-11-08 15:57:59 ·
1053 阅读 ·
1 评论