tinyflow源码阅读-0

来源:互联网 发布:股票辅助软件 编辑:程序博客网 时间:2024/06/03 05:28

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?

原创粉丝点击