当前位置: 首页 > 工具软件 > TinyFlow > 使用案例 >

tinyflow源码阅读-0

敖和韵
2023-12-01

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?

 类似资料: