2021-06-15 记事

2021-06-15 记事 在js上运行PyTorch模型 ONNX.js ------- Run PyTorch models in the browser using ONNX.js https://github.com/elliotwaite/pytorch-to-javascript-with-onnx-js 微软还有一个项目 …

torch模型导出到onnx问题 expected 1 dims but tensor ha…

导出过程中出现如下错误 ``` expected 1 dims but tensor has 2 ``` torch 1.5.0 must be used for the export because the following error occurs intorch 1.6.0 caused by the internal optimization…

onnx runtime fastapi模型部署

fastapi 模型部署大体流程如下： * torch-> onnx * fastapi做为接口服务，后端使用onnx推理 * Docker打包部署 fastapi官…