目前已经实现MLP网络的搭建和训练,bin文件夹中提供的是linux环境下的预编译程序
具体实现在aedlf.cpp中,目前因时间安排问题,暂无编写相关说明的意向,等到TODO适当完结后进行相关补充
核心代码均在include文件夹中,src文件夹中可能残留了开发早期代码,但无法使用
因本项目目前定位偏向学习性,因此矩阵计算库也为自己实现的(参考include/math),计算库相关的HPC操作也会列入TODO范畴
- 静态计算图自动搭建
- 自动求导
- 简单的数据载入(依靠numpy)
- 调试CV相关算子
- 编写优化器相关代码
- 优化计算过程和计算库
- 切换到动态图
- 增加CUDA加速相关功能
- 支持计算图剪枝优化等操作
- 支持模型并行、数据并行