【DAY8-代码】Pytorch缝合模块之Resnet和SE注意力模块，以及修改模型后如何加载预训练权重

准备代码

首先把这篇文章的代码copy下来。【DL系列】ResNet网络结构详解、完整代码实现

也可以直接看resnet源码（复杂一点）。resnet源码

再看SE注意力模块代码注意力机制（一）SE模块（Squeeze-and-Excitation Networks）论文总结和代码实现

把SE模块复制到resnet的model文件中，塞到某个子模块下面就行。注意，basicblock和bottleneck都是resnet网络的子模块，ResNet才是主网络。

#全局平均池化+1*1卷积核+ReLu+1*1卷积核+Sigmoid
class SE_Block(nn.Module):
    def __init__(self, inchannel