Revert CE fix (#1740)

4a8bea03 · qingqing01 · GitHub · 44786e9c · 4a8bea03
隐藏空白更改
内联并排

Showing with 2 addition and 3 deletion

PaddleNLP/sequence_tagging_for_ner/train.py PaddleNLP/sequence_tagging_for_ner/train.py +2 -3

未找到文件。
--- 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)