Alexnet的基础知识

本文讨论了过拟合问题,介绍了如何通过dropout技术减少神经网络的过拟合,并给出了一个PyTorch模型的前向传播函数示例,涉及特征提取、平均池化和分类过程。
摘要由CSDN通过智能技术生成

过拟合,拟合的函数完美的预测了训练集,但对新数据的测试集预测结果差,泛化能力弱

dropout

在网络正向传播时随机失活一部分神经元

减少过拟合,减少训练参数

经卷积后的矩阵尺寸大小计算公式

n = (w-f+2*p)/(s+1)

1,输入图片大小w*w

2,filter大小f*f

3,步长s

4,padding的像素级p

n*n图像经过f*f的卷积核后得到的图像大小为((n-f)+1)*((n-f)+1)

padding =0 ,表示不填充,padding=1,表示在边界填充一个像素值,padding-2表示填充2个像素值

def forward(self, x): x = self.features(x) x = self.avgpool(x) x = torch.flatten(x, 1) x = self.classifier(x) return x

这是一个PyTorch模型的前向传播函数。它首先通过self.features(x)对输入x进行特征提取,然后通过self.avgpool(x)对特征进行平均池化,接着使用torch.flatten(x, 1)将池化后的特征展平,最后通过self.classifier(x)对展平后的特征进行分类。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值