PyTorch笔记 - A ConvNet for the 2020s (ConvNeXt) 网络

Posted 2023-03-03 SpikeKing

欢迎关注我的CSDN：https://blog.csdn.net/caroline_wendy
本文地址：https://blog.csdn.net/caroline_wendy/article/details/128236899

Paper：A ConvNet for the 2020s，FAIR，Berkeley
Code：https://github.com/facebookresearch/ConvNeXt

网络结构：

• stem layer：骨干层，目的是降低计算复杂度。
• Why does CNNs usually have a stem?

实现ConvNeXt的Block，类似于ResNet，只不过是d7x7(depth-wise) -> LN -> 1x1(通道融合，提升通道) -> 1x1(通道融合，降低通道)

Block源码：继承于nn.Module