from torch import nn


## new loss functions here


## and add them to the loss_funcs dictionary here

loss_funcs = {
    "mse": nn.MSELoss(),
    "huber": nn.HuberLoss(),
    "l1": nn.L1Loss(),
    "l2": nn.MSELoss(),  # Alias for mse
    "smooth_l1": nn.SmoothL1Loss(),
    "cross_entropy": nn.CrossEntropyLoss(),
    "bce": nn.BCEWithLogitsLoss(),  # Binary Cross Entropy with Logits
}
