更新2020-02-12

  • 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

猜你喜欢

转载自www.cnblogs.com/LuckBelongsToStrugglingMan/p/12300080.html