INNER CODE UNIT · Python

lr

ShannonAI/glyce · glyce/models/latticeLSTM/main.py:152

    lr = init_lr * ((1-decay_rate)**epoch)
    logger.info(F" Learning rate is setted as: {lr}")
    for param_group in optimizer.param_groups:
        param_group['lr'] = lr
    return optimizer


def evaluate(data, model, name):
    if name == "train":
        instances = data.train_Ids
    elif name == "dev":
        instances = data.dev_Ids
    elif name == 'test':
        instances = data.test_Ids
    elif name == 'raw':
        instances = data.raw_Ids
    else:
        logger.info("Error: wrong evaluate name," + name)

View source record →