k510例程模型结构和自己训练的模型结构不一样导致错误

这是例程模型的结构


这是我根据手册训练出来的自己的

结构不一样导致用起来的时候报错,我应该在那里进行修改呢?
又或者说剪枝这一步应该在哪里完成?一定要剪成例程这种结构吗?

需要剪枝成我们提供的示例模型结构,由于开发板端不支持yolov5后处理的加速