exp.py 文件源码

def get_criterion(exp_name):
    '''
    create loss function based on parameters loaded from config
    '''

    cfg = config.load_config_file(exp_name)

    criterion_name = cfg['criterion']
    loss_method = getattr(loss, criterion_name)
    criterion = loss_method()

    return criterion