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,6 +136,7 @@ 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)