自动混合精度(AMP)训练在低版本显卡上的使用问题

AMP training on NVIDIA GeForce GTX 1660 SUPER GPU may cause NaN losses or zero-mAP results, so AMP will be disabled during training.

这个警告提示表明在NVIDIA GeForce GTX 1660 SUPER显卡上使用自动混合精度(AMP)训练可能导致损失变为NaN或mAP结果为零,因此训练过程中将自动禁用AMP。以下是对该问题的详细分析和解决方案:

你可能感兴趣的:(Pytorch,人工智能)