Caffe2 近日在其博客上公布了与 PyTorch 合并的各项细节,文中表示 Caffe2 的开发重点是性能和跨平台部署,而 PyTorch 则专注于快速原型设计和研究的灵活性。二者的组件在过去一年大量被共享,双方也意识到将各自的优势特性整合到一个包中,并实现从快速原型到快速执行的平稳过渡是有意义的,也能通过更轻松地使用共享工具提高开发效率。
Caffe2 和 PyTorch 的代码已于 4 月开始合并,在接下来的几个月里,还将继续深入整合框架的各个组件,并将它们作为一个整体进行有效整合。最终将 PyTorch 前端的灵活用户体验与 Caffe2 后端的扩展、部署和嵌入功能相结合。
合并细节主要包括:
brew
,core.Net
)将继续工作,并且现有的序列化模型 NetDefs 会提供向后兼容性。