PyTorch源码解读之torchvision.models
发现一篇写得很好的博客,码一下:
PyTorch源码解读之torchvision.models
补充几个作者没有写的点:
1.inplanes表示输入的通道数,因为resnet中每个block会改变channel数,但几个相同的block串联的时候,会导致channel数合不上,inplanes用于记录当前block的输入channel数,当block的实际输入channel数和所需的...
原创
2018-10-28 16:01:25 ·
648 阅读 ·
0 评论