首页 > 其他 > 详细

更新2020-02-12

时间:2020-02-12 18:43:51      阅读:63      评论:0      收藏:0      [点我收藏+]
  • model.named_children() 与 model.named_modules()的意思:
    • 目的:返回网络model下的模块,并返回名称和模块本身
    • 区别:.named_children()返回所有子模块,并返回名称和模块本身;.named_modules()返回所有模块(包括子模块下的所有模块),并返回名称和模块本身。
    • 相同点:两者都是返回模块迭代器,只是前者将子模块作为最小单元返回,后者将每层作为最小单元返回。
    • 例子:代码可参阅:https://blog.csdn.net/watermelon1123/article/details/98036360
  • hook: pytorch 可以利用register_forward_hook和register_backward_hook方法将网络的特定模块提取出来,并能够返回相应模块的输入和输出。这两个方法分别用于输出特定模块的features/gradients,为可视化中间层特征提供了帮助。参见:半小时学会Pytorch Hookhttps://zhuanlan.zhihu.com/p/75054200

更新2020-02-12

原文:https://www.cnblogs.com/LuckBelongsToStrugglingMan/p/12300080.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!