多卡训练的wts生成engine会偶发报错

[TinkingLoeng]

使用官方的YOLOv5n生成engine成功

使用自己训练的YOLOv5n生成engine失败（类别数已做修改）

紧接着通过多次重复实验，发现YOLOV5n v6.0版本的代码，在将多卡训练得到的wts转成engine时会偶发失败，报错如上图。但同样的参数在单卡训练得到的wts基础上进行转换则可成功转成engine

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

[wang-xinyu]

@TinkingLoeng Maybe there is some difference between the .pt models. You need to look into it.

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.