mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

rename_train_result

pull/3294/head
zhangyubo0722 2024-11-13 06:06:36 +00:00
parent 4e2e0fe51f
commit 5172a2fc66
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/utils/save_result.py
View File

@ -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"

4
tools/train.py
View File

@ -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)