diff --git a/ppocr/modeling/architectures/base_model.py b/ppocr/modeling/architectures/base_model.py
index 03fbcee8465df9c8bb7845ea62fc0ac04917caa0..ff3da01a95c1d2f63cf00a4cf09b41b8b37b1c87 100644
--- a/ppocr/modeling/architectures/base_model.py
+++ b/ppocr/modeling/architectures/base_model.py
@@ -79,7 +79,10 @@ class BaseModel(nn.Layer):
             x = self.neck(x)
         y["neck_out"] = x
         x = self.head(x, targets=data)
-        y["head_out"] = x
+        if type(x) is dict:
+            y.update(x)
+        else:
+            y["head_out"] = x
         if self.return_all_feats:
             return y
         else: