作者您好，请问我用train_prune_sfp训练出来的pt应该怎么转成onnx呢

[wyzwyzzzzz]

我用yolov5 3.0的代码中的expert.py好像并不能转成onnx，会出现错误

[xhwNobody]

我用yolov5 3.0的代码中的expert.py好像并不能转成onnx，会出现错误

麻烦把你的错误贴出来呢

[wyzwyzzzzz]

感谢您的回答，我把训练出来的网络直接转到v3.1的工程下，执行export.py，就出现了这个错误，出现在experimental.py中，请问应该怎么解决呢（我训练是采用的原来读取yolov5l.yaml的方法）

[xhwNobody]

感谢您的回答，我把训练出来的网络直接转到v3.1的工程下，执行export.py，就出现了这个错误，出现在experimental.py中，请问应该怎么解决呢（我训练是采用的原来读取yolov5l.yaml的方法）

我看这个报错只是简单的加载模型出错问题呀

[wyzwyzzzzz]

感谢您的回答，我把训练出来的网络直接转到v3.1的工程下，执行export.py，就出现了这个错误，出现在experimental.py中，请问应该怎么解决呢（我训练是采用的原来读取yolov5l.yaml的方法）

我看这个报错只是简单的加载模型出错问题呀

感谢您的再次回答， 但是我换用工程自带的yolov5s的模型就可以转换成功，3.0和3.1的代码都有试过，请问您有没有试过将pt转成onnx呢，是怎么转的呢

我感觉可能是export.py中attempt_load用的是models里面的，但是您的模型构建是用的model.py里面的，我刚尝试了import model之后，它的报错就变成了 'Model' object has no attribute 'fuse'

在我尝试把.fuse()给删掉之后，出现了新的报错，还是找不到model

[xhwNobody]

感谢您的回答，我把训练出来的网络直接转到v3.1的工程下，执行export.py，就出现了这个错误，出现在experimental.py中，请问应该怎么解决呢（我训练是采用的原来读取yolov5l.yaml的方法）

我看这个报错只是简单的加载模型出错问题呀

感谢您的再次回答， 但是我换用工程自带的yolov5s的模型就可以转换成功，3.0和3.1的代码都有试过，请问您有没有试过将pt转成onnx呢，是怎么转的呢

我感觉可能是export.py中attempt_load用的是models里面的，但是您的模型构建是用的model.py里面的，我刚尝试了import model之后，它的报错就变成了 'Model' object has no attribute 'fuse'

在我尝试把.fuse()给删掉之后，出现了新的报错，还是找不到model

不好意思，最近有空看了一下，发现了出现这个问题可以尝试加上 import sys,os sys.path.append('../') 对比一下get_small_model.py就知道啦