一、问题现象（：
yolov5-PyTorch的loss收敛速度比官方慢很多

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

三、测试步骤：
使用Yolov5 version6的Modelzoo代码训练，参数与本地GPU上全部一致，发现loss收敛速度比本地慢很多，性能在相同epoch情况下也有所下降。

具体而言，cls-loss对比本地跑的一直降不下去
经检查，该仓库下的Yolo的loss.py写法与官方并不一致
我试图修改为官方的用法会出现loss计算nan的问题。
请问能否阐明本仓库写法与官方有何不同，能否给出官方代码的适配方法