caffe訓練時出現的錯誤

(1)loss值居高不下,一直是87.3365 解決方法:將訓練網絡中BatchNorm中的use_global_stats值改爲  false如下所示(注意要全部改完) (2)訓練階段出現如下所示錯誤: 原因:batch_size太大,導致內存不足,我的顯卡是gtx1650,因此batch_size分別設置爲100和20才得以跑通。
相關文章
相關標籤/搜索