torch.nn.utils.clip_grads_with_norm_¶
- torch.nn.utils.clip_grads_with_norm_(parameters, max_norm, total_norm, foreach=None)[source]¶
根据预先计算的总范数和期望的最大范数,缩放参数迭代器的梯度。
梯度将通过以下计算进行缩放
梯度会被就地修改。
此函数等效于预先计算总范数的
torch.nn.utils.clip_grad_norm_()
。
根据预先计算的总范数和期望的最大范数,缩放参数迭代器的梯度。
梯度将通过以下计算进行缩放
梯度会被就地修改。
此函数等效于预先计算总范数的 torch.nn.utils.clip_grad_norm_()
。