Spec-Zone.ru › PyTorch 1

torch.nn.utils.clip_grad_value_

torch.nn.utils.clip_grad_value_(parameters, clip_value) [source]

Ограничивает градиент итерируемого набора параметров заданным значением.

Градиенты изменяются на месте.

Параметры:
  • parameters (Iterable[Tensor] или Tensor) – итерируемый набор тензоров или один тензор, для которого будут нормализованы градиенты
  • clip_value (float или int) – максимальное разрешенное значение градиентов. Градиенты ограничиваются диапазоном [-clip_value,clip_value]\left[\text{-clip\_value}, \text{clip\_value}\right]

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.utils.clip_grad_value_.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API