compute_gradient_norm¶
-
class
maze.train.utils.train_utils.
compute_gradient_norm
(params: Iterable[torch.Tensor])¶ Computes the cumulative gradient norm of all provided parameters.
- Parameters
params – Iterable over model parameters.
- Returns
The cumulative gradient norm.