BatchNorm 动量约定 PyTorch

2024-04-26

Is the 批归一化动量约定 http://pytorch.org/docs/master/_modules/torch/nn/modules/batchnorm.html(默认=0.1)与其他库一样正确,例如Tensorflow默认情况下似乎通常是0.9或0.99?或者也许我们只是使用不同的约定?


看来pytorch中的参数化约定与tensorflow中的不同,因此pytorch中的0.1相当于tensorflow中的0.9。

更准确地说:

在张量流中:

running_mean = decay*running_mean + (1-decay)*new_value

在 PyTorch 中:

running_mean = (1-decay)*running_mean + decay*new_value

这意味着值为decay在 PyTorch 中相当于一个值(1-decay)在张量流中。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

BatchNorm 动量约定 PyTorch 的相关文章

随机推荐