torch.nn.utils.
clip_grad_value_
Clips gradient of an iterable of parameters at specified value.
Gradients are modified in-place.
parameters (Iterable[Tensor] or Tensor) – an iterable of Tensors or a single Tensor that will have gradients normalized
clip_value (float or int) – maximum allowed value of the gradients. The gradients are clipped in the range [-clip_value,clip_value]\left[\text{-clip\_value}, \text{clip\_value}\right][-clip_value,clip_value]
此页内容是否对您有帮助
感谢反馈!