当前位置:   article > 正文

.pt转onnx报错TorchScript export failure_pt转onnx 模型载入不成功

pt转onnx 模型载入不成功

出现上述报错
解决办法:在转换指令:python export.py --grid --end2end --simplify --topk-all 100 --iou-thres 0.3 --conf-thres 0.8 --img-size 640 640 --max-wh 640 --weights weights/onnxruntime.pt
取掉其中的:–grid --end2end --simplify

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小惠珠哦/article/detail/850165
推荐阅读
相关标签
  

闽ICP备14008679号