From 4a8bea037f0e2640e2dac4085a238a93b6ec427d Mon Sep 17 00:00:00 2001 From: qingqing01 Date: Sat, 2 Feb 2019 13:46:45 +0800 Subject: [PATCH] Revert CE fix (#1740) --- PaddleNLP/sequence_tagging_for_ner/train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/PaddleNLP/sequence_tagging_for_ner/train.py b/PaddleNLP/sequence_tagging_for_ner/train.py index 68e62137..b77c081b 100644 --- a/PaddleNLP/sequence_tagging_for_ner/train.py +++ b/PaddleNLP/sequence_tagging_for_ner/train.py @@ -136,9 +136,8 @@ def main(train_data_file, " pass_f1_score:" + str(test_pass_f1_score)) save_dirname = os.path.join(model_save_dir, "params_pass_%d" % pass_id) - if "CE_MODE_X" not in os.environ: - fluid.io.save_inference_model(save_dirname, ['word', 'mark'], - crf_decode, exe) + fluid.io.save_inference_model(save_dirname, ['word', 'mark'], + crf_decode, exe) if "CE_MODE_X" in os.environ: print("kpis train_precision %f" % pass_precision) -- GitLab