fix bug when load dict for optimizer in resume training

2022-05-17 21:04:01 +08:00 · 2022-05-17 21:04:01 +08:00 · ec80a56e5c
parent f4e400fa3a
commit ec80a56e5c
1 changed files with 2 additions and 1 deletions
--- a/ppcls/utils/save_load.py
+++ b/ppcls/utils/save_load.py
@ -105,7 +105,8 @@ def init_model(config, net, optimizer=None, loss: paddle.nn.Layer=None):
        net.set_state_dict(para_dict)
        loss.set_state_dict(para_dict)
        for i in range(len(optimizer)):
-            optimizer[i].set_state_dict(opti_dict[i])
+            optimizer[i].set_state_dict(opti_dict[i] if isinstance(
+                opti_dict, list) else opti_dict)
        logger.info("Finish load checkpoints from {}".format(checkpoints))
        return metric_dict