caffe2源代码归入pytorch:caffe2与pytorch模型相互转化

Song • 1141 次浏览 • 0 个回复 • 2018年04月02日

2018-04-01日,caffe2官方github页面发布公告,caffe2源代码现已归入pytorchFacebook两大深度学习框架合二为一;其实这两个框架有很多共享的代码和相同的库,合并之后,能减少同时维护两个深度学习框架的成本。

Caffe2是一个轻量级的、模块化的、可扩展的深度学习框架。在原Caffe的基础上,Caffe2的设计思想主要是表达,速度,和模块化。 caffe2源代码现已归入pytorch

caffe2自从20171月发布之后,由于调试、编译等多方面的优势,PyTorch已经成为很多科研机构首选的深度学习框架;而20174月推出的 Caffe 2则具有可在iOSAndroid和树莓派等多种设备上训练和部署模型的优势。尽管获得了很多用户的支持,在面对谷歌支持的TensorFlow生态时,PyTorchCaffe 2各自仍有短板,此次「合并」或许会成为深度学习工程领域新形势的一个开始。

PyTorch最近几个月有一些改变:

  • PyTorchCaffe2目前会共享CI,这是非常重要的工程工作。
  • PyTorchPyTorch-ONNX有非常复杂的CIonnxbot触发器构建在每一个PyTorch PR上,并以roundabout的方式更新。
  • 「pending」状态中有后端研发工作,例如与最新和最重要的库集成(MKLDNN、cuFFT和更多的NNPACK覆盖等)。

作为PyTorch老用户,其实安装搭建和使用没有什么变化,你可以像以前一样使用pytorch和开发的拓展。

关于caffe2源代码现已归入pytorch,目前任Facebook研究科学家贾扬清在知乎上表示:

因为PyTorch有优秀的前端,Caffe2有优秀的后端,整合起来以后可以进一步最大化开发者的效率。目前FAIR大概有超过一半的项目在使用PyTorch,而产品线全线在使用Caffe2,所以两边都有很强的动力来整合优势。

开发效率是我在Facebook非常重视的一个方向:去年年中的时候启动了ONNX项目(初版的代码是我亲自上手写的),然后帮助搭建了ONNX team,来增强不同框架甚至不同公司之间的协作;Caffe2PyTorch在代码层的合并也是从那个时候开始逐渐推动的一项内容。

至于进一步的计划,目前我还不方便透露,等过一个月有空再来更新吧。

部分参考:机器之心


原创文章,转载请注明 :caffe2源代码归入pytorch:caffe2与pytorch模型相互转化 - pytorch中文网
原文出处: https://ptorch.com/news/148.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! Pytorch中集成BeamSearch算法的中文聊天机器人