fix ema resume (#5413)

f3f8dc9c · wangxinxin08 · GitHub · 1d8dc38f · f3f8dc9c
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppdet/optimizer.py ppdet/optimizer.py +2 -1

未找到文件。
--- a/ppdet/optimizer.py
+++ b/ppdet/optimizer.py
@@ -339,7 +339,8 @@ class ModelEMA(object):

    def resume(self, state_dict, step=0):
        for k, v in state_dict.items():
-            self.state_dict[k] = v
+            if k in self.state_dict:
+                self.state_dict[k] = v
        self.step = step

    def update(self, model=None):