
前言
接前面的博客说,如要调用自己训练好的模型,需要修改以下参数。
一、export.py的参数修改
在models中找到export.py文件
修改第24行的权重默认路径
配置时需要加绝对路径 否则不成功 然后根据之前的博客导出ONNX文件在调用就可以了。
欢迎分享,转载请注明来源:内存溢出

接前面的博客说,如要调用自己训练好的模型,需要修改以下参数。
在models中找到export.py文件
修改第24行的权重默认路径
配置时需要加绝对路径 否则不成功 然后根据之前的博客导出ONNX文件在调用就可以了。
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)