【发布时间】: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