From f3f8dc9cbdf6b0b7ba3fc758b692233658e413a6 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Mon, 28 Mar 2022 20:59:00 +0800
Subject: [PATCH] fix ema resume (#5413)

---
 ppdet/optimizer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppdet/optimizer.py b/ppdet/optimizer.py
index fd46c949a..a0a961b45 100644
--- 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):
-- 
GitLab