pytorch eval作用

发表于 | 阅读次数:

锁住batch_norm层

1
2
3
During training, this layer keeps a running estimate of its computed mean and variance. The running sum is kept with a default momentum of 0.1.

During evaluation, this running mean/variance is used for normalization.

请作者喝一杯咖啡☕️