快速上手-pytorch 使用BN,Drop,权重初始化,mini-batch等方式

本文快速介绍如何在PyTorch中应用BN层、Dropout正则化以及权重初始化来降低过拟合。BN用于批量归一化,Dropout在训练时启用以防止过拟合,权重初始化通常采用normal_分布。此外,通过Dataloader设置mini-batch大小进行训练。实验证明,这些方法能有效提升模型性能。
摘要由CSDN通过智能技术生成

简介

本文只针对快速上手,BN,Dropout,权重初始化等降低过拟合的方法请自行百度

BN

batch norm
在init中初始化BN,1d针对一维、二维输入(加上第一维的batchsize 所以最终输入是2-3维)
2d、3d同理:

self.bn = nn.BatchNorm1d(x特征维度)

然后向前传播中加在输入后即可,

x = self.bn(x)

Dropout

和BN一样 初始化:

 self.dropout = nn
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值