if __name__ == "__main__":
    loss_fn = AsymmetricMSELoss(under_prediction_weight=3.0)
    under_prediction_loss = loss_fn(torch.tensor([0.0]), torch.tensor([1.0]))
    over_prediction_loss = loss_fn(torch.tensor([2.0]), torch.tensor([1.0]))