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

迪巴拉 发表于 2024-8-6 15:57:27

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

TyCoding 发表于 2024-8-6 15:58:03

提示词爱用不用

Crystαl 发表于 2024-8-6 15:59:02

什么意思

拾光发表于 2024-8-6 15:59:39

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

TyCoding 发表于 2024-8-6 15:59:44

谢谢，我试试，netron 是不是不能查看pt

页: [1]

DZ插件网's Archiver