INNER CODE UNIT · Python

load_model_decode

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

def load_model_decode(save_dir, data):
    logger.info("Load Model from file: " + save_dir)
    model = SeqModel(data)
    model.load_state_dict(torch.load(save_dir))
    logger.info(F"Decode dev data ...")
    start_time = time.time()
    speed, acc, p, r, f, pred_results = evaluate(data, model, 'dev')
    end_time = time.time()
    time_cost = end_time - start_time
    logger.info(("%s: time:%.2fs, speed:%.2fst/s; acc: %.4f, p: %.4f, r: %.4f, f: %.4f"%('dev', time_cost, speed, acc, p, r, f)))

    logger.info(F"Decode test data ...")
    start_time = time.time()
    speed, acc, p, r, f, pred_results = evaluate(data, model, 'test')
    end_time = time.time()
    time_cost = end_time - start_time
    logger.info(("%s: time:%.2fs, speed:%.2fst/s; acc: %.4f, p: %.4f, r: %.4f, f: %.4f"%('test', time_cost, speed, acc, p, r, f)))

View source record →