PyTorch中的BatchNorm2d层

先来看看pytorch中对于类的定义:python 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)中
相关文章
相关标签/搜索