caffe2
RiweiChen
talk is cheap, show me the code
展开
-
caffe2实现多任务学习
caffe2实现多任务原创 2017-07-27 01:24:49 · 3822 阅读 · 0 评论 -
caffe2 mdl文件转init_net.pb, predict_net.pb
前言caffe2中,完整的训练例子就train_resnet50.py, 其模型和snapshot都保持在同一个文件中,不像caffe 一样有.caffemodel 和 .snapshot 文件(其实是冗余的)。训练完之后发现*.mdl文件不能直接使用,需要转化为两个pb文件,然而没有说怎么转换,也有很多人在问这个issue,然而并没有人说出正确的方法。解决方法通过看代码(好像也没有别的方法了),发原创 2017-07-27 22:28:27 · 3630 阅读 · 1 评论