InfiniTensor/pyinfinitensor
YdrMaster c9fee3f667 feat: 前端支持 GlobalAveragePool 及单元测试
Signed-off-by: YdrMaster <ydrml@hotmail.com>
2023-02-16 10:33:24 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor feat: 前端支持 GlobalAveragePool 及单元测试 2023-02-16 10:33:24 +08:00
tests feat: 前端支持 GlobalAveragePool 及单元测试 2023-02-16 10:33:24 +08:00
README.md feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
pyproject.toml feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00

README.md

pyinfinitensor

InfiniTensor 的 Python 前端。

  • 从 ONNX 导入模型
  • 从 Pytorch 导入模型
  • 从 PaddlePaddle 导入模型
  • 模型导出到 ONNX
  • 模型导出到 Pytorch
  • 模型导出到 PaddlePaddle

python 工程结构及打包方法

本项目使用 [pyproject.toml] 文件定义,目录结构采用 src 布局,后端库会被打包,并支持自动的依赖安装

参考 setuptools 的文档