unilab.algos.torch.common.stability.clip_gradients

unilab.algos.torch.common.stability.clip_gradients(parameters, max_norm=10.0)[source]

Clip gradients by global norm.

Parameters:

  • parameters – Model parameters

  • max_norm (float) – Maximum gradient norm