2018年1月10日
caffe 错误
摘要： 一些caffe错误 训练时很快梯度爆炸，loss猛增至nan 如果找不到数据上的原因的话，可以怀疑caffe框架有问题，换用其它版本试试。比如我遇到的问题是在训练时使用了Accuracy层，而该层的实现代码在某次更新中GPU代码存在bug，复用了其它层的变量导致对loss的计算产生了影响。训练时去掉 阅读全文
posted @ 2018-01-10 11:00 康行天下 阅读(990) 评论(0) 推荐(0) 编辑