Update train.py (#1733)

5e37cabc · kolinwei · Yibing Liu · 0eda9630 · 5e37cabc
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

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

未找到文件。
--- a/PaddleNLP/sequence_tagging_for_ner/train.py
+++ b/PaddleNLP/sequence_tagging_for_ner/train.py
@@ -136,8 +136,9 @@ 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)
-        fluid.io.save_inference_model(save_dirname, ['word', 'mark'],
-                                      crf_decode, exe)
+        if "CE_MODE_X" not in os.environ:
+            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)