From 9711fe5ea8611f386bd9a6b721709ea481fa008e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Wed, 23 Aug 2023 22:51:29 +0800 Subject: [PATCH] Update utils.py --- utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 12e0a42..fab8ce7 100644 --- a/utils.py +++ b/utils.py @@ -22,7 +22,8 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False learning_rate = checkpoint_dict['learning_rate'] if optimizer is not None and not skip_optimizer and checkpoint_dict['optimizer'] is not None: optimizer.load_state_dict(checkpoint_dict['optimizer']) - elif optimizer is None and not skip_optimizer: + elif optimizer is None and not skip_optimizer: #Disable this line if Infer ,and enable the line below + #else: new_opt_dict = optimizer.state_dict() new_opt_dict_params = new_opt_dict['param_groups'][0]['params'] new_opt_dict['param_groups'] = checkpoint_dict['optimizer']['param_groups']