nvidia版caffe訓練的模型對bvlc版的兼容性

需要在solver.prototxt 文件中設置:

store_blobs_in_old_format: true

參考:https://docs.nvidia.com/deeplearning/dgx/caffe-user-guide/index.html#storeblobsold

另外訓練速度方面,nvidia版caffe快50%(單卡)到3倍(8卡)左右。

 

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