From ba6df9cfaa80c3c8d5c2f63b0fbd550ab908d4b9 Mon Sep 17 00:00:00 2001 From: will-jl944 Date: Mon, 11 Oct 2021 11:33:17 +0800 Subject: [PATCH] fix resume checkpoint bug --- paddlex/utils/checkpoint.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddlex/utils/checkpoint.py b/paddlex/utils/checkpoint.py index 5b974306d9..978cbf567f 100644 --- a/paddlex/utils/checkpoint.py +++ b/paddlex/utils/checkpoint.py @@ -424,6 +424,9 @@ def load_pretrain_weights(model, pretrain_weights=None, model_name=None): def load_optimizer(optimizer, state_dict_path): logging.info("Loading optimizer from {}".format(state_dict_path)) optim_state_dict = paddle.load(state_dict_path) + for key in optimizer.state_dict().keys(): + if key not in optim_state_dict.keys(): + optim_state_dict[key] = optimizer.state_dict()[key] if 'last_epoch' in optim_state_dict: optim_state_dict.pop('last_epoch') optimizer.set_state_dict(optim_state_dict)