记录bug:使用Tensorflow kersa加载模型出错

错误内容

  File "/root/anaconda3/envs/python36c/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 1516, in load_weights
    saving.load_weights_from_hdf5_group(f, self.layers)
  File "/root/anaconda3/envs/python36c/lib/python3.6/site-packages/tensorflow/python/keras/engine/saving.py", line 772, in load_weights_from_hdf5_group
    original_keras_version = f.attrs['keras_version'].decode('utf8')
AttributeError: 'str' object has no attribute 'decode'

解决方法

         使用2.10版本的h5py包。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple h5py==2.10

参考资料：

original_keras_version = f.attrs[‘keras_version‘].decode(‘utf8‘)_乐亦亦乐的博客-CSDN博客