From d92aa33c618337f5ebe538a2db612fa5a6e95854 Mon Sep 17 00:00:00 2001 From: wangxiao1021 Date: Wed, 12 Feb 2020 10:57:53 +0800 Subject: [PATCH] fix tagging --- examples/tagging/run.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/tagging/run.py b/examples/tagging/run.py index f39d8a0..c99dc70 100644 --- a/examples/tagging/run.py +++ b/examples/tagging/run.py @@ -65,13 +65,13 @@ if __name__ == '__main__': trainer.fit_reader(seq_label_reader) # # step 8-1*: load pretrained parameters - # trainer.load_pretrain(pre_params) + trainer.load_pretrain(pre_params) # # step 8-2*: set saver to save model save_steps = 1951 # print('save_steps: {}'.format(save_steps)) - # trainer.set_saver(save_path=save_path, save_steps=save_steps, save_type=save_type) + trainer.set_saver(save_path=save_path, save_steps=save_steps, save_type=save_type) # # step 8-3: start training - # trainer.train(print_steps=train_print_steps) + trainer.train(print_steps=train_print_steps) # ----------------------- for prediction ----------------------- @@ -95,7 +95,7 @@ if __name__ == '__main__': # step 6: load pretrained model pred_model_path = './outputs/ckpt.step' + str(save_steps) - pred_ckpt = trainer.load_ckpt(pred_model_path) + trainer.load_ckpt(pred_model_path) # step 7: fit prepared reader and data trainer.fit_reader(predict_seq_label_reader, phase='predict') -- GitLab