torchsig.models.spectrogram_models.detr.criterion.sigmoid_ce_loss¶
- torchsig.models.spectrogram_models.detr.criterion.sigmoid_ce_loss(inputs: Tensor, targets: Tensor, num_masks: float)[source]¶
- Parameters:
inputs – A float tensor of arbitrary shape. The predictions for each example.
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).
- Returns:
Loss tensor