【问题标题】:load_weights_from_hdf5_group original_keras_version AttributeError: 'str' object has no attribute 'decode'load_weights_from_hdf5_group original_keras_version AttributeError:“str”对象没有属性“decode”
【发布时间】:2021-10-30 02:28:14
【问题描述】:

尝试加载保存的权重

robertamodel = TFAutoModelForSequenceClassification.from_pretrained("roberta-base",num_labels=6)
loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
metric = tf.keras.metrics.SparseCategoricalAccuracy('0accuracy')
optimizer = tf.keras.optimizers.Adam(learning_rate=1e-5,epsilon=1e-08)

robertamodel.compile(loss=loss,optimizer=optimizer,metrics=[metric])
robertamodel.load_weights(r"/media/sh/robert_model6classes.h5")

我得到错误

load_weights_from_hdf5_group original_keras_version

AttributeError: 'str' 对象没有属性 'decode'

【问题讨论】:

    标签: python keras load bert-language-model


    【解决方案1】:

    试试这个,因为这对我也有帮助:

    pip uninstall h5py
    pip install h5py==2.10.0
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2018-12-01
      • 2014-05-26
      • 2021-07-05
      • 1970-01-01
      • 1970-01-01
      • 2015-05-15
      • 2021-07-06
      • 2019-11-02
      相关资源
      最近更新 更多