From ec80a56e5cae27decf30c97d26e52262363c07e8 Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Tue, 17 May 2022 21:04:01 +0800
Subject: [PATCH] fix bug when load dict for optimizer in resume training

---
 ppcls/utils/save_load.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py
index 7e7869d20..1f3c66d45 100644
--- 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