Skip to content

Commit a18f9f5

Browse files
cocuhfchollet
authored andcommitted
fix character type in loading weights. (#5762)
1 parent 7f58b6f commit a18f9f5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

keras/engine/topology.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2839,11 +2839,11 @@ def load_weights_from_hdf5_group(f, layers):
28392839
and weights file.
28402840
"""
28412841
if 'keras_version' in f.attrs:
2842-
original_keras_version = f.attrs['keras_version']
2842+
original_keras_version = f.attrs['keras_version'].decode('utf8')
28432843
else:
28442844
original_keras_version = '1'
28452845
if 'backend' in f.attrs:
2846-
original_backend = f.attrs['backend']
2846+
original_backend = f.attrs['backend'].decode('utf8')
28472847
else:
28482848
original_backend = None
28492849

@@ -2911,11 +2911,11 @@ def load_weights_from_hdf5_group_by_name(f, layers):
29112911
and weights file.
29122912
"""
29132913
if 'keras_version' in f.attrs:
2914-
original_keras_version = f.attrs['keras_version']
2914+
original_keras_version = f.attrs['keras_version'].decode('utf8')
29152915
else:
29162916
original_keras_version = '1'
29172917
if 'backend' in f.attrs:
2918-
original_backend = f.attrs['backend']
2918+
original_backend = f.attrs['backend'].decode('utf8')
29192919
else:
29202920
original_backend = None
29212921

0 commit comments

Comments
 (0)