class BatchNorm(in_channels, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
参数:
in_channels (int) – 输入通道大小.
eps (float, optional) – 为分母增加数值的稳定性. (default: 1e-5)
momentum (float, optional) – 用于运行平均值和运行方差计算的值. (default: 0.1)
affine (bool, optional) – 如果设置为True，则此模块具有可学习的仿射参数γ 和 β. (default: True)
track_running_stats (bool, optional) – 如果设置为True，则此模块跟踪运行平均值和方差； 设置为False，此模块不跟踪此类统计信息，并且始终在训练和评估模式下使用批处理统计信息。（ (default: True)
原创文章 32
获赞 4
访问量 7629