bnm_validator

BNMValidator

Bases: SimpleLossValidator

Returns the negative of the BNM loss of all logits.

Source code in pytorch_adapt\validators\bnm_validator.py

class BNMValidator(SimpleLossValidator):
    """
    Returns the negative of the
    [BNM loss][pytorch_adapt.layers.bnm_loss.BNMLoss]
    of all logits.
    """

    @property
    def loss_fn(self):
        return BNMLoss()