8-resnet 手写

nutron-ma

已于 2023-11-16 17:03:13 修改

阅读量253

点赞数

分类专栏：深度学习pytorch系列文章文章标签：深度学习 python 人工智能

于 2022-12-14 15:24:19 首次发布

本文链接：https://blog.csdn.net/qq_43328166/article/details/128317025

版权

深度学习pytorch系列文章专栏收录该内容

22 篇文章 3 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客详细介绍了如何构建并应用手写的8-resnet残差模块，深入探讨了深度学习中的这一关键组件。

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

残差模块示例：
在这里插入图片描述

'''Residual Block残差模块'''
class ResidualBlock(nn.Module):
    def __init__(self,in_channel=64,out_channel=64,k=3,s=1):
        super(ResidualBlock, self).__init__()
        self.conv = nn.Sequential(
            nn.Conv2d(in_channel, out_channel, kernel_size=k, stride=s, padding=1),
            nn.BatchNorm2d(in_channel),
            nn.PReLU(),   # 自动学习斜率的LeakyReLU
            nn.Conv2d(in_channel, out_channel, kernel_size=k, stride=s, padding=1),
            nn.BatchNorm2d(in_channel)
        )
    def forward(self,x):
        x = x + self.conv(x)   # 残差精华&#x

了解本专栏