tinyflow简介——从tqchen/tinyflow翻译而来
TinyFlow是NNVM的示例代码,旨在展示如何用少量的代码(2K行)构建自己的深度学习系统。构建的DL系统基于图计算,具有TensorFlow式的API。其中,运算符由Torch7实现,而更侧重于展示系统概念。
TinyFlow可以运行在CPU和GPU上,代码结构如下:
- 927行运算符代码(C++中嵌入Lua)
- 734行运算执行代码
- 71行API接口
- 233行前段代码
为什么写TinyFlow?
- DL系统教学材料,tqchen开设了个课程,CSE 599G1: Deep Learning System
- 通过NNVM快速自定义DL系统,试验新特性
- 验证中间表示层(如NNVM)对多种前段(TensorFlow,MxNet)和后端(Torch,MxNet)的支撑能力
- 不同DL系统中模块重用的试验田
- 好玩
知乎上也有一些讨论,如何评价陈天奇的模块化深度学习系统NNVM?