BatchNorm动量约定PyTorch

pet*_*eld 10 python neural-network deep-learning pytorch batch-normalization

batchnorm势头约定(默认值= 0.1),正确的,因为在其他的库,例如Tensorflow这似乎通常是在默认情况下为0.9或0.99?或许我们只是使用不同的约定?

pat*_*_ai 12

似乎pytorch中的参数化约定与tensorflow中的参数化约定不同,因此pytorch中的0.1相当于张量流中的0.9.

更确切地说:

在Tensorflow中:

running_mean = decay*running_mean + (1-decay)*new_value
Run Code Online (Sandbox Code Playgroud)

在PyTorch中:

running_mean = (1-decay)*running_mean + decay*new_value
Run Code Online (Sandbox Code Playgroud)

这意味着decayPyTorch中的值等于Tensorflow中的值(1-decay).