torchsig.models.spectrogram_models.detr.criterion.dice_loss

torchsig.models.spectrogram_models.detr.criterion.dice_loss(inputs: Tensor, targets: Tensor, num_masks: float)

Compute the DICE loss, similar to generalized IOU for masks :param inputs: A float tensor of arbitrary shape.

The predictions for each example.

Parameters:

targets –

A float tensor with the same shape as inputs. Stores the binary

classification label for each element in inputs

(0 for the negative class and 1 for the positive class).