BN中use_global_stats问题

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/csuzhaoqinghui/article/details/84944892

BN层中有个参数use_global_stats,它表示是否使用caffe内部的均值和方差。 
训练模型的时候,将BN层use_global_stats设置为false;测试的时候设置为true,不然训练的时候会报nan或者模型不收敛。

猜你喜欢

转载自blog.csdn.net/csuzhaoqinghui/article/details/84944892
BN