您的位置:首页 > 其它

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

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

解决办法:这个问题是tensorlayer旧版本中的bug,将tensorlayer更新到最新版本1.4.1后,该问题就解决了。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐