BatchNorm2d
Tchunren
本人喜欢运动,热爱生活
展开
-
pytorch nn.BatchNorm2d() 参数解释
在使用pytorch的 nn.BatchNorm2d() 层的时候,经常地使用方式为在参数里面只加上待处理的数据的通道数(特征数量),但是有时候会在后面再加入一个小数,比如这样 nn.BatchNorm2d(64,0.8),这里面的0.8有什么作用呢? 我们知道在训练过程中 nn.BatchNorm2d() 的作用是根据统计的mean 和var来对数据进行标准化,并且这...原创 2020-03-19 09:29:59 · 32583 阅读 · 5 评论 -
在pytorch模型中如何获得BatchNorm2d层的各个mean和var(平均值和方差)
这个内容是将随便做了一个网络结构,然后简单的训练几次,生成模型,并且存储起来,主要是为了学习获得pytorch中的BatchNorm2d层的各个特征图的平均值和方差。代码如下:import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torchvision import datasets,transform...原创 2019-12-13 20:38:17 · 3542 阅读 · 0 评论