加载model一直报错unexpected backbone.bn1.num_batches_tracked
原来是使用的版本的问题,pytorch0.4.1之后在BN层加入了track_running_stats这个参数
这个参数的作用如下:
训练时用来统计训练时的forward过的min-batch数目,每经过一个min-batch, track_running_stats+=1
如果没有指定momentum, 则使用1/num_batches_tracked 作为因数来计算均值和方差(running mean and variance).
参考:https://blog.csdn.net/shanglianlm/article/details/101394508
backbone.bn1.num_batches_tracked 报错
最新推荐文章于 2023-11-04 18:00:14 发布