先来看看pytorch中对于类的定义:
CLASS torch.nn.BatchNorm2d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
解析一下每个参数的作用:
num_features:BatchNorm是针对每一个通道做的,所以这里应该填写(N, C, H, W)中的C
eps: 在对参数进行标准化的时候,需要保证不出现除零错误,故需要eps参数
momentum :这里,E[x]与Var[x]不仅仅与本batch的x有关,而是采用了一个滑动平均来全面考虑了之前所有batch的情况。滑动平均的参数即为momentum
affine:进行放射变换,如果为False则