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

fix code

pull/1853/head
HydrogenSulfate 2022-04-20 11:08:42 +08:00
parent 3a1276d315
commit 3f117428da
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/optimizer/__init__.py
View File

@ -61,7 +61,7 @@ def build_optimizer(config, epochs, step_each_epoch, model_list=None):
for optim_item in optim_config: for optim_item in optim_config:
# optim_cfg = {optim_name1: {scope: xxx, **optim_cfg}} # optim_cfg = {optim_name1: {scope: xxx, **optim_cfg}}
# step1 build lr # step1 build lr
optim_name = optim_item.keys()[0] # get optim_name1 optim_name = list(optim_item.keys())[0] # get optim_name1
optim_scope = optim_item[optim_name].pop('scope') # get scope optim_scope = optim_item[optim_name].pop('scope') # get scope
optim_cfg = optim_item[optim_name] # get optim_cfg optim_cfg = optim_item[optim_name] # get optim_cfg

2
ppcls/utils/save_load.py
View File

@ -103,7 +103,7 @@ def init_model(config, net, optimizer=None, loss: paddle.nn.Layer=None):
metric_dict = paddle.load(checkpoints + ".pdstates") metric_dict = paddle.load(checkpoints + ".pdstates")
# set state dict # set state dict
net.set_state_dict(para_dict) net.set_state_dict(para_dict)
loss.loss_func[i].set_state_dict(para_dict) loss.set_state_dict(para_dict)
for i in range(len(optimizer)): for i in range(len(optimizer)):
optimizer[i].set_state_dict(opti_dict) optimizer[i].set_state_dict(opti_dict)
logger.info("Finish load checkpoints from {}".format(checkpoints)) logger.info("Finish load checkpoints from {}".format(checkpoints))