diff --git a/ppcls/utils/save_result.py b/ppcls/utils/save_result.py index f7613db36..5714676ca 100644 --- a/ppcls/utils/save_result.py +++ b/ppcls/utils/save_result.py @@ -49,7 +49,7 @@ def update_train_results(config, assert last_num >= 1 train_results_path = os.path.join(config["Global"]["output_dir"], - "train_results.json") + "train_result.json") save_model_tag = ["pdparams", "pdopt", "pdstates"] save_inference_tag = [ "inference_config", "pdmodel", "pdiparams", "pdiparams.info" diff --git a/tools/train.py b/tools/train.py index fa6e36bc6..2912e3e91 100644 --- a/tools/train.py +++ b/tools/train.py @@ -37,11 +37,11 @@ if __name__ == "__main__": if uniform_output_enabled: if os.path.exists( os.path.join(config["Global"]["output_dir"], - "train_results.json")): + "train_result.json")): try: os.remove( os.path.join(config["Global"]["output_dir"], - "train_results.json")) + "train_result.json")) except: pass config_dict = convert_to_dict(config)