INNER CODE UNIT · Python

train

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

def train(data, save_model_dir, seg=True):
    logger.info("Training model...")
    data.show_data_summary()
    model = SeqModel(data)
    logger.info("finished built model.")
    parameters = [p for p in model.parameters() if p.requires_grad]
    optimizer = optim.SGD(parameters, lr=data.HP_lr, momentum=data.HP_momentum)
    best_dev = -1
    data.HP_iteration = 100
    #  start training
    for idx in range(data.HP_iteration):
        epoch_start = time.time()
        temp_start = epoch_start
        logger.info(("Epoch: %s/%s" %(idx,data.HP_iteration)))
        optimizer = lr_decay(optimizer, idx, data.HP_lr_decay, data.HP_lr)
        instance_count = 0
        sample_loss = 0
        batch_loss = 0

View source record →