ResNet解析(pytorch源码)

首先放一张各层的图片,整体分为4个layer, pytorch中也是这么分的 然后这是两种设计方式,左边的是用于18,34层的,这样参数多,右面这种设计方式参数少,适用于更深度的 这里是这两个基本块的代码,然后ResNet中把这些块连接起来就可以组成网络。最重要的代码就是下面这3部分,然后通过传入不同的参数给ResNet就可以建立各种经典层的网络了。 class BasicBlock(nn.Mod
相关文章
相关标签/搜索