forked from jiuyuan/InfiniTensor
YdrMaster
341cf1f943
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 的文档。