yolo v3训练过程中报错 non-finite loss的解决方法

报错

WARNING: non-finite loss, ending training {'box_loss': tensor([nan], device='cuda:0', grad_fn=<MulBackward0>), 'obj_loss': tensor([nan], device='cuda:0', grad_fn=<MulBackward0>), 'class_loss': tensor([nan], device='cuda:0', grad_fn=<MulBackward0>)}

解决方法

如果你是windows10环境下的pytorch1.8+cuda11
重新安装 pytorch1.7.1+cuda10.1即可完美解决。