yolo预训练模型导出为onnx格式报错

Question

yolov5添加了psp分割头，使用export.py转化预训练模型为onnx格式报错，具体报错如图

Answer 1

你是不是把预训练模型的参数放到一个list里面了？
建议检查代码内容，把list里面的内容取出来，合并转化为tensor格式。