ResNet解析(pytorch源码)

首先放一张各层的图片,总体分为4个layer, pytorch中也是这么分的python 而后这是两种设计方式,左边的是用于18,34层的,这样参数多,右面这种设计方式参数少,适用于更深度的网络 这里是这两个基本块的代码,而后ResNet中把这些块链接起来就能够组成网络。最重要的代码就是下面这3部分,而后经过传入不一样的参数给ResNet就能够创建各类经典层的网络了。app class Basic
相关文章
相关标签/搜索