tensorlayer舊版本(1.3以下版本)中,predict過程也會更新BatchNormLayer的參數

問題描述:我的tensorlayer版本是1.2.8,訓練模型中使用了BatchNormLayer層,訓練好模型後,在預測過程中,利用predict函數來循環預測多張圖片的label,發現上一張圖片的預測過程會對下一張圖片的預測結果造成影響。後面發現原來預測完一張圖片後,模型中batchNormLayer層的參數會發生變化。


解決辦法:這個問題是tensorlayer舊版本中的bug,將tensorlayer更新到最新版本1.4.1後,該問題就解決了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章