ggt.losses.aleatoric_loss

Module Contents

Functions

aleatoric_loss(outputs, targets[, average])

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