torchdms.loss.weighted_loss

torchdms.loss.weighted_loss(base_loss)[source]

Generic loss function decorator with loss decay or target exponentiation.