mindspore训练yolov5时,loss=nan

1、使用modelzoo下载的yolo5在gpu训练时,loss出现nan的情况,请问一下一般出现nan的原因是啥,如何排查?

下图是训练的打印信息,里面有具体的参数

mindspore训练yolov5时,loss=nan_第1张图片

 

上面问题建议参考这里排查下问题

精度调优 — MindSpore master documentation

mindspore训练yolov5时,loss=nan_第2张图片

 

你可能感兴趣的:(python,深度学习,tensorflow)