bugfix of optimizer resume (PaddlePaddle#129)

qizhaoaoe · Oct 14, 2022 · 5202eaf · 5202eaf
1 parent e30861d
commit 5202eaf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plsc/optimizer/optimizer.py b/plsc/optimizer/optimizer.py
@@ -178,7 +178,9 @@ def cast(param, value):
         for k, v in state_dict['state'].items():
             if k in param_name_map:
                 param = param_name_map[k]
-                state[param] = cast(param, v)
+                # TODO(GuoxiaWang): fix type cast
+                # state[k] = cast(param, v)
+                state[k] = v
             else:
                 state[k] = v