pytorch基础教程-模型编写(二)

对于一些重复的layer可以采用nn.Sequential先进行封装layer,然后调用_make_layer_进行实现 以ResNet18/34为例 模型图: ResNet中将一个跨层直连的单元叫做Residual block,其结构如下,当输入和输出的通道不一致的时候,需要一个专门的单元将二者转成一致,使其可以相加。 实现: class BasicBlock(nn.Module): e
相关文章
相关标签/搜索