发现一篇写得很好的博客,码一下:
PyTorch源码解读之torchvision.models
补充几个作者没有写的点:
1.inplanes表示输入的通道数,因为resnet中每个block会改变channel数,但几个相同的block串联的时候,会导致channel数合不上,inplanes用于记录当前block的输入channel数,当block的实际输入channel数和所需的输入channel数不一致时,会加入一个downsample。
2.expansion表示经过一个block后,channel是原来的几倍。