Pytorch残差网络ResNet

Jokic_Rn

已于 2022-06-21 14:59:24 修改

阅读量527

点赞数

文章标签： pytorch 网络深度学习

于 2022-02-14 17:29:00 首次发布

本文链接：https://blog.csdn.net/weixin_44823313/article/details/122928370

版权

本文介绍了在Pytorch中实现残差网络ResNet的结构，并提供了完整的训练代码链接，尽管存在一些泛化问题，作者计划后续进行优化。

摘要由CSDN通过智能技术生成

Pytorch残差网络ResNet

完整训练代码见 >>Github链接，泛化结果有一点点问题，之后有时间再改

ResNet结构

在这里插入图片描述

class Residual(nn.Module):
    def __init__(self,input_channels,num_channels,use_1x1conv=False,strides=1):
        super().__init__()
        self.conv1 = nn.Conv2d(input_channels,num_channels,
                               kernel_size=3,padding=1,stride=strides)
        self.conv2 = nn.Conv2d(num_channels,num_channels,
                               kernel_size=3,padding=1)
        if use_1x1conv:
            self.conv3 = nn.</