caffe训练时出现的错误

(1)loss值居高不下,一直是87.3365 解决方法:将训练网络中BatchNorm中的use_global_stats值改为  false如下所示(注意要全部改完) (2)训练阶段出现如下所示错误: 原因:batch_size太大,导致内存不足,我的显卡是gtx1650,因此batch_size分别设置为100和20才得以跑通。
相关文章
相关标签/搜索