【亲测免费】 建立自己的深度学习系统：TinyFlow，仅需2000行代码！

TinyFlow是一个小巧且强大的深度学习框架，它基于DMLC的NNVM项目，旨在展示如何构建一个与TensorFlow接口相同，但更简洁、基础的计算图系统。这个项目不仅可运行于GPU和CPU上，而且其源码量小，易于理解和学习，是教学、研究和实验的理想平台。

项目介绍

在TinyFlow中，你可以找到927行用于实现操作符的代码，734行执行运行时代码，71行API胶水代码以及233行前端代码。尽管体积小巧，但它包含了自动微分、形状类型推断、静态内存分配等先进特性，足以支持复杂的训练和推理任务。此外，它还利用Torch7作为默认后端，使得快速实现操作符变得简单，同时也展现了将Lua嵌入到C++中的乐趣。

技术分析

TinyFlow的设计灵感来自于Unix哲学，即通过模块化和可重用性来简化开发流程。它的API自动复用了NNVM，选择了Torch7作为运算执行后台，这使我们能够专注于系统的构造，而不是操作符的编写。虽然没有选择MXNet作为前端或后端（因为MXNet已经使用了NNVM作为中间层），但这带来了新的尝试和挑战。

此外，TinyFlow支持自动求导，可以在编译阶段进行形状和类型的推断，并引入静态内存分配以实现高效的训练和推理。这一切都封装在一个仅2000行代码的系统内。

应用场景

• 教学材料：对于教授深度学习系统概念，TinyFlow是一个极好的起点，可以设置关于符号微分、内存分配和操作符融合的作业。
• 研究实验：研究人员可以轻松地在TinyFlow中添加新特性，由于其模块化设计，这些改进也可以移植到其他使用NNVM的系统。
• IR案例展示：演示如何利用中间表示如NNVM来实现跨前端（TF, MXNet）和后端（Torch7, MXNet）的多目标优化。
• 通用模块测试床：与其他系统（如MXNet）一起，TinyFlow可以作为测试共享深度学习系统通用组件的平台。

项目特点

• 精简高效：少于2000行代码即可实现一个完整的深度学习系统。
• 易扩展：模块化设计允许快速添加新功能。
• 兼容性好：与Torch7集成，能复用其丰富的运算符库。
• 先进的特性：自动微分、形状和类型推断以及静态内存管理。
• 教育价值：适用于深度学习系统课程的教学。

安装与运行

首先安装Torch7，然后设置环境变量TORCH_HOME指向Torch的根目录。接下来，确保NNVM已安装并配置了融合插件，最后修改TinyFlow的Makefile启用USE_FUSION，然后编译。通过设置PYTHONPATH包括tinyflow和nnvm的Python路径，你可以运行示例程序，例如python example/mnist_softmax.py或example/mnist_lenet.py。

让我们一起探索TinyFlow的世界，发掘深度学习系统的奥秘，并享受自定义和构建你的深度学习系统的乐趣！