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