利用GPU训练,loss异常
收藏
我在使用gpu时候,loss值和我在windows利用cpu训练差距较大,平台上训练会出现
2019-02-21T09:06:46.183692185Z Pass[0];Iter[10]; train loss: 0.123; mean loss: 30807495471529984.000 2019-02-21 09:06:46
2019-02-21T09:07:05.097732303Z Pass[0];Iter[11]; train loss: 24200437758427136.000; mean loss: 30206850737635328.000 2019-02-21 09:07:05
2019-02-21T09:07:24.295951173Z Pass[0];Iter[12]; train loss: 2288818405244928.000; mean loss: 27880348557770752.000 2019-02-21 09:07:24
这些较大的loss值,windows利用cpu训练不会出现
0
收藏
请登录后评论
还请提供一下源码