yolov5的loss收敛速度比官方慢很多

[asdfo123]

一、问题现象（附报错日志上下文）：
yolov5 version6, version7的loss收敛速度都比官方慢很多

二、软件版本:
--CANN 版本: CANN 6.3.2
--Pytorch版本: 1.11.0
--Python 版本：3.7.10
--操作系统版本: ubuntu_20.04-x86_64

三、测试步骤：
使用Yolov5 version6, version7的Modelzoo-GPL代码训练，参数与本地GPU上全部一致，发现loss收敛速度比本地慢很多。
经检查，该仓库下的Yolo的loss.py写法与官方并不一致
我试图修改为官方的用法会出现loss计算nan的问题。
请问能否阐明本仓库写法与官方有何不同，能否给出官方代码的适配方法