unilab.algos.torch.common.stability.check_nan_loss

unilab.algos.torch.common.stability.check_nan_loss(loss, default_metrics)[source]

Check if loss contains NaN or Inf values.

Parameters:

  • loss (Tensor) – Loss tensor to check

  • default_metrics (dict) – Default metric values to return if NaN detected

Return type:

tuple[Tensor | None, dict | None]

Returns:

(loss, None) if valid, (None, nan_metrics) if invalid