rename_train_result

Author: zhangyubo0722

paddleseg/utils/save_info.py

@@ -37,7 +37,7 @@ def update_train_results(args,
                          ema=False):
     assert last_num >= 1
     cfg = Config(args.config)
-    train_results_path = os.path.join(args.save_dir, "train_results.json")
+    train_results_path = os.path.join(args.save_dir, "train_result.json")
     save_model_tag = ["pdparams", "pdopt", "pdstates"]
     save_inference_tag = [
         "inference_config", "pdmodel", "pdiparams", "pdiparams.info"

tools/train.py

@@ -183,8 +183,8 @@ def main(args):
     if uniform_output_enabled:
         if not os.path.exists(args.save_dir):
             os.makedirs(args.save_dir)
-        if os.path.exists(os.path.join(args.save_dir, "train_results.json")):
-            os.remove(os.path.join(args.save_dir, "train_results.json"))
+        if os.path.exists(os.path.join(args.save_dir, "train_result.json")):
+            os.remove(os.path.join(args.save_dir, "train_result.json"))
         with open(os.path.join(args.save_dir, "config.yaml"), "w") as f:
             yaml.dump(cfg.dic, f)
     print_mem_info = cfg.dic.pop('print_mem_info', True)