ggt.losses.aleatoric_loss
Module Contents
Functions
|
Computes the aleatoric loss. |
- ggt.losses.aleatoric_loss.aleatoric_loss(outputs, targets, average=True)
Computes the aleatoric loss. Args:
outputs: (tensor) - predicted outputs from the model targets: (tensor) - ground truth labels size_average: (bool) - if True, the losses are
averaged over all elements of the batch
- Returns:
aleatoric_loss: (tensor) - aleatoric loss