批归一化BN层总结

今天总结一下之前学习的批归一化层也就是Batch Normalize层。 PyTorch中的BN层: 在PyTorch的torch.nn模块中提供三种归一化操作,分别用于不同的输入数据: BatchNorm1d(num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True) 一般用于输入数据是,由2d数据组成
相关文章
相关标签/搜索