simbert训练踩坑记2

报错：Traceback (most recent call last):
  File "stage2.py", line 206, in <module>
    roformer.load_weights("./best_model_2.weights")
  File "/output/.pylibs/lib/python3.6/site-packages/keras/engine/saving.py", line 492, in load_wrapper
    return load_function(*args, **kwargs)
  File "/output/.pylibs/lib/python3.6/site-packages/keras/engine/network.py", line 1230, in load_weights
    f, self.layers, reshape=reshape)
  File "/output/.pylibs/lib/python3.6/site-packages/keras/engine/saving.py", line 1183, in load_weights_from_hdf5_group
    original_keras_version = f.attrs['keras_version'].decode('utf8')
AttributeError: 'str' object has no attribute 'decode'

该问题是缺失或者h5py版本不对，一句话就能解决

pip install h5py==2.8.0