yolov5中SGD换Adam后result.png结果很乱的问题

我在metrics.py中加入了EIoU后，并且SGD换Adam后，results.png就变成下面这样。

 

 在metrics.py中改变下顺序即可，我将这三句靠在一起就解决了。

with torch.no_grad():
    alpha = v / (v - iou + (1 + eps))
return iou - (rho2 / c2 + v * alpha)  # CIoU

如图：

问题解决了。