serotiny.nn.losses.continuous_bernoulli module

class serotiny.nn.losses.continuous_bernoulli.CBLogLoss(reduction='mean', mode='probs')

Bases: _Loss

Continuous Bernoulli loss, proposed here:

https://arxiv.org/abs/1907.06845.

forward(input, target)

reduction: str