import torch


def loss_get(args):
    choice_dict = {'bce': 'torch.nn.BCEWithLogitsLoss()','cross': 'torch.nn.CrossEntropyLoss()'}
    loss = eval(choice_dict[args.loss])
    return loss