[bugfix] fix erniedoc (#8393)

w5688414 · web-flow · commit 9381e9fe7821 · 2024-05-17T16:10:50.000+08:00
* fix erniedoc

* revert
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__":

Original file line number	Diff line number	Diff line change
`@@ -337,8 +337,8 @@ def do_train(args):`
`337`	`337`	`)`
`338`	`338`	`# Copy the memory`
`339`	`339`	`memories = create_memory()`
`340`		`- predict(model, test_dataloader, args.file_path, memories, test_ds.label_list)`
`341`		`- logger.info("Done Predicting the results has been saved in file: {}".format(args.file_path))`
	`340`	`+ predict(model, test_dataloader, args.test_results_file, memories, test_ds.label_list)`
	`341`	`+ logger.info("Done Predicting the results has been saved in file: {}".format(args.test_results_file))`
`342`	`342`
`343`	`343`
`344`	`344`	`if __name__ == "__main__":`