Function build_loss

program.py:113–116

(flags)

def build_loss(flags):
    loss_params = flags.Loss.function
    loss = create_module(loss_params)(params=flags.Loss)
    return loss


def build_trainer(model, optimizer, loss, train_loader, val_loader, \