forked from jiuyuan/InfiniTensor
YdrMaster
296fcc5aa0
- python 前端项目结构及打包和安装脚本 - 后端编译出 so 改名为 backend，增加 GraphHandler 修改图结构 - ci 支持测试这些功能 Signed-off-by: YdrMaster <ydrml@hotmail.com> |
||
|---|---|---|
| .. | ||
| docs | ||
| src/pyinfinitensor | ||
| tests | ||
| README.md | ||
| pyproject.toml | ||
README.md
pyinfinitensor
InfiniTensor 的 Python 前端。
- 从 ONNX 导入模型
- 从 Pytorch 导入模型
- 从 PaddlePaddle 导入模型
- 模型导出到 ONNX
- 模型导出到 Pytorch
- 模型导出到 PaddlePaddle
python 工程结构及打包方法
本项目使用 [pyproject.toml] 文件定义，目录结构采用 src 布局，后端库会被打包，并支持自动的依赖安装。
参考 setuptools 的文档。