bert4keras icon indicating copy to clipboard operation
bert4keras copied to clipboard

加载的模型无法直接调用么

Open iRonaldo opened this issue 3 years ago • 3 comments

bert_encoder = build_transformer_model( checkpoint_path=checkpoint_path, config_path=config_path, return_keras_model=True) hidden_states_1,_ = bert_encoder([inputs[0], inputs[1]]) hidden_states_2,_ = bert_encoder([inputs[2], inputs[3]])

想用两个输入通过同一个bert形成两个输出 这一步一直过不去

报错 Tensor objects are only iterable when eager execution is enabled. To iterate over this tensor use tf.map_fn.

iRonaldo avatar May 27 '21 06:05 iRonaldo

麻烦按照提问模版提供详细的信息。

bojone avatar May 27 '21 06:05 bojone

哦,对了,不知道你莫名其妙地加个,_干嘛,bert_encoder返回的是一个tensor,你强行映射为两个,这是你的这个错误的原因。

bojone avatar May 27 '21 06:05 bojone

感谢回复, 这是在keras-bert, bert4keras来回倒腾的结果 我把tf114换成tf2.3.1貌似可以直接用()调用bert_encoder得到想要的结果, docker上gpu用tf 114才没有问题,太惨了 114还是有问题

On May 27, 2021, at 2:46 PM, 苏剑林(Jianlin Su) @.***> wrote:

哦,对了,不知道你莫名其妙地加个,_干嘛,bert_encoder返回的是一个tensor,你强行映射为两个,这是你的这个错误的原因。

iRonaldo avatar May 27 '21 06:05 iRonaldo