[bugfix] fix erniedoc (#8393)

* fix erniedoc * revert
PaddlePaddle · May 17, 2024 · 9381e9f · 9381e9f
1 parent d9dcd9a
commit 9381e9f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/text_classification/ernie_doc/train.py b/examples/text_classification/ernie_doc/train.py
@@ -337,8 +337,8 @@ def do_train(args):
     )
     # Copy the memory
     memories = create_memory()
-    predict(model, test_dataloader, args.file_path, memories, test_ds.label_list)
-    logger.info("Done Predicting the results has been saved in file: {}".format(args.file_path))
+    predict(model, test_dataloader, args.test_results_file, memories, test_ds.label_list)
+    logger.info("Done Predicting the results has been saved in file: {}".format(args.test_results_file))
 
 
 if __name__ == "__main__":