INNER CODE UNIT · Python

load_data_setting

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

def load_data_setting(save_dir):
    with open(save_dir, 'rb') as fp:
        data = torch.load(fp)
    logger.info("Data setting loaded from file: " + save_dir)
    data.show_data_summary()
    return data


def lr_decay(optimizer, epoch, decay_rate, init_lr):
    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":

View source record →