keras-bert,加载预训练模型报错 Layer model_1 expects 3 inputs, but it received 2 input tensors


问题:在使用keras-bert 导入预训练的模型时, 报错Layer model_1 expects 3 inputs, but it received 2 input tensors
导入代码:
bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path, training=True,output_layer_num=7,trainable=True,seq_len=Config.max_seq_len)  # 加载预训练模型

具体报错:
Layer model_1 expects 3 inputs, but it received 2 input tensors. Input received: [<tf.Tensor 'input_1:0' shape=(?, 100) dtype=float32>, <tf.Tensor 'input_2:0' shape=(?, 100) dtype=float32>]

解决方式:去掉参数 training=True
具体:bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path, output_layer_num=7, trainable=True, seq_len=Config.max_seq_len)

 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM