近来在训练检测网络的时候会出现loss为nan的情况，需要中断重新训练，会很麻烦。因而选择使用pytorch提供的梯度裁剪库来对模型训练过程中的梯度范围进行限制，修改之后，不再出现loss为nan的情况。
pytorch中采用torch.nn.utils.clip_grad_norm_来实现梯度裁剪，链接如下：
训练代码使用示例如下：
其中，max_norm为梯度的最大范数，也是梯度裁剪时主要设置的参数。
备注：网上有同学提醒在（强化学习）使用了梯度裁剪之后训练时间会大大增加。目前在我的检测网络训练中暂时还没有碰到这个问题，以后遇到再来更新。
补充：pytorch训练过程中出现nan的排查思路
1、最常见的就是出现了除0或者log0这种
看看代码中在这种操作的时候有没有加一个很小的数，但是这个数数量级要和运算的数的数量级要差很多。一般是1e-8。
2、在optim.step()之前裁剪梯度
max_norm一般是1，3，5。
3、前面两条还不能解决nan的话
就按照下面的流程来判断。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持www.887551.com。
黄山市民网：https://www.huangshanshimin.com/
