Fix yaml version compat issue

5be4ed3d · François Chollet · f2bbf98b · 5be4ed3d
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

keras/engine/saving.py keras/engine/saving.py +4 -1

未找到文件。
--- a/keras/engine/saving.py
+++ b/keras/engine/saving.py
@@ -639,7 +639,10 @@ def model_from_yaml(yaml_string, custom_objects=None):
    # Returns
        A Keras model instance (uncompiled).
    """
-    config = yaml.load(yaml_string, Loader=yaml.FullLoader)
+    if hasattr(yaml, 'FullLoader'):
+        config = yaml.load(yaml_string, Loader=yaml.FullLoader)
+    else:
+        config = yaml.load(yaml_string)
    from ..layers import deserialize
    return deserialize(config, custom_objects=custom_objects)