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

Rename variable

pull/2840/head
Bobholamovic 2023-06-13 20:39:58 +08:00 committed by cuicheng01
parent d6137854e2
commit b4f10436cf
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppcls/utils/save_load.py
View File

@ -165,8 +165,8 @@ def save_model(net,
return
if prefix == 'best_model':
uapi_best_model_path = os.path.join(model_path, 'best_model')
_mkdir_if_not_exist(uapi_best_model_path)
best_model_path = os.path.join(model_path, 'best_model')
_mkdir_if_not_exist(best_model_path)
_mkdir_if_not_exist(model_path)
model_path = os.path.join(model_path, prefix)
@ -188,8 +188,8 @@ def save_model(net,
paddle.save(params_state_dict, model_path + ".pdparams")
if prefix == 'best_model':
uapi_best_model_path = os.path.join(uapi_best_model_path, 'model')
paddle.save(params_state_dict, uapi_best_model_path + ".pdparams")
best_model_path = os.path.join(best_model_path, 'model')
paddle.save(params_state_dict, best_model_path + ".pdparams")
if ema is not None:
paddle.save(ema.state_dict(), model_path + ".ema.pdparams")