写在前面
最近开始尝试了caffe2,主要是因为tensorflow的发展越来越奇怪,一种网络实现在tensorflow中有多种对应的实现方式,臃肿的函数表达,每月一次大更新没有向下兼容,越来越高昂的学习成本,加之最近其他几大深度学习框架对tensorflow有合围之势…太多的槽点了,让我有点想放弃tensorflow这个坑了。还是怀念caffe的简洁,于是还是捡起贾扬清大神的续集填坑吧。废话不多说了,还是看看caffe2的亮点吧
caffe2亮点
- 支持分布式训练
- 对移动端的deployment的支持
- 基础计算单元改为Operators,相对于caffe的layers单元有很多改进
caffe模型导入caffe2
python -m caffe2.python.caffe_translator deploy.prototxt pretrained.caffemodel
torch模型导入caffe2
使用torch2caffe的工具,将caffe作为桥梁,不展开讨论了