fix character type in loading weights. (#5762)

a18f9f57 · Kosuke Kusano · François Chollet · 7f58b6fb · a18f9f57
隐藏空白更改
内联并排

Showing with 4 addition and 4 deletion

keras/engine/topology.py keras/engine/topology.py +4 -4

未找到文件。
--- a/keras/engine/topology.py
+++ b/keras/engine/topology.py
@@ -2839,11 +2839,11 @@ def load_weights_from_hdf5_group(f, layers):
            and weights file.
    """
    if 'keras_version' in f.attrs:
-        original_keras_version = f.attrs['keras_version']
+        original_keras_version = f.attrs['keras_version'].decode('utf8')
    else:
        original_keras_version = '1'
    if 'backend' in f.attrs:
-        original_backend = f.attrs['backend']
+        original_backend = f.attrs['backend'].decode('utf8')
    else:
        original_backend = None

@@ -2911,11 +2911,11 @@ def load_weights_from_hdf5_group_by_name(f, layers):
            and weights file.
    """
    if 'keras_version' in f.attrs:
-        original_keras_version = f.attrs['keras_version']
+        original_keras_version = f.attrs['keras_version'].decode('utf8')
    else:
        original_keras_version = '1'
    if 'backend' in f.attrs:
-        original_backend = f.attrs['backend']
+        original_backend = f.attrs['backend'].decode('utf8')
    else:
        original_backend = None