完整的ResNet代码( 四 )


所以可以进行一个总结:
函数用来制作残差块的结构,参数layer可以用来控制每个残差层是由多少个残差块组成的,在残差块结构中判断是否采用(1x1卷积进行升维)是根据步长或者输入输出通道数是否相等,如果步长为1,输入通道数不等于输出通道数就会采用一个1x1卷积进行升维 。每个残差层layer只有第一个残差块是采用了 。
后续将结合这一部分做知识蒸馏的讲解,请持续关注