x as dict

ppocr/modeling/architectures/base_model.py

@@ -79,6 +79,9 @@ class BaseModel(nn.Layer):
             x = self.neck(x)
         y["neck_out"] = x
         x = self.head(x, targets=data)
+        if type(x) is dict:
+            y.update(x)
+        else:
             y["head_out"] = x
         if self.return_all_feats:
             return y