嘟嘟社区

调用现有的AI模型,是只要知道 输入和输出有什么就行么


是拿到模型后,只需要知道输入格式是什么,按照这个格式传数据进去就行么?
调了几天pytorch的好像是这样。
另外问下有什么查看模型有什么输入输出的工具
和模型格式转换工具,把pytorch的转成onnx
提示词爱用不用

jqbaobao 发表于 2024-8-6 11:13
提示词爱用不用

什么意思

https://netron.app/
输入,输出都要跟模型的入参和出参匹配。
实际就是相应大小的数据流的输入和输出。