BN层中有个参数use_global_stats,它表示是否使用caffe内部的均值和方差。
训练模型的时候,将BN层use_global_stats设置为false;测试的时候设置为true,不然训练的时候会报nan或者模型不收敛。
BN中use_global_stats问题
最新推荐文章于 2022-07-19 12:07:51 发布
BN层中有个参数use_global_stats,它表示是否使用caffe内部的均值和方差。
训练模型的时候,将BN层use_global_stats设置为false;测试的时候设置为true,不然训练的时候会报nan或者模型不收敛。