🐛 Fix multi-label model load issue.

BrikerMan · Nov 18, 2020 · 4ee0f3c · 4ee0f3c
1 parent 7f87bdc
commit 4ee0f3c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kashgari/utils/serialize.py b/kashgari/utils/serialize.py
@@ -26,8 +26,9 @@ def load_data_object(data: Dict,
     """
     if custom_objects is None:
         custom_objects = {}
+
     if data['__class_name__'] in custom_objects:
-        obj: Any = custom_objects[data['__class_name__']](**kwargs)
+        obj: Any = custom_objects[data['__class_name__']](**data['config'], **kwargs)
     else:
         module_name = f"{data['__module__']}.{data['__class_name__']}"
         obj: Any = pydoc.locate(module_name)(**data['config'], **kwargs)  # type: ignore