美文网首页
Pytorch 之 正则化 BatchNorm2d()函数的使用

Pytorch 之 正则化 BatchNorm2d()函数的使用

作者: Allard_c205 | 来源:发表于2021-08-18 15:57 被阅读0次

bn层的位置位于卷积神经网络的卷积层之后,对数据进行归一化处理,避免数据在进行下一步处理(卷积或激活函数)时因数据过大造成过拟合,导致网络性能不稳定,bn的函数数学原理如下:

bn的参数有:

num_features:输入参数是batch_size*num_features*height*width,即为其中特征的数量。

bn的结果就是代表卷积层数据经过正则化层之后得到的网络层数据

相关文章

网友评论

      本文标题:Pytorch 之 正则化 BatchNorm2d()函数的使用

      本文链接:https://www.haomeiwen.com/subject/rcvgbltx.html