def get_criterion()

in svhn_based_exp/utils.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

def get_criterion(task):
    if task == "lf-mis-up":
        criterion = torch.nn.MSELoss()
    else:
        criterion = nn.CrossEntropyLoss()
    return criterion