tensorlayer旧版本(1.3以下版本)中,predict过程也会更新BatchNormLayer的参数

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

问题描述:我的tensorlayer版本是1.2.8,训练模型中使用了BatchNormLayer层,训练好模型后,在预测过程中,利用predict函数来循环预测多张图片的label,发现上一张图片的预测过程会对下一张图片的预测结果造成影响。后面发现原来预测完一张图片后,模型中batchNormLayer层的参数会发生变化。


解决办法:这个问题是tensorlayer旧版本中的bug,将tensorlayer更新到最新版本1.4.1后,该问题就解决了。

猜你喜欢

转载自blog.csdn.net/feimengjuan/article/details/67631902