bert4keras
bert4keras copied to clipboard
加载的模型无法直接调用么
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.
麻烦按照提问模版提供详细的信息。
哦,对了,不知道你莫名其妙地加个,_
干嘛,bert_encoder
返回的是一个tensor,你强行映射为两个,这是你的这个错误的原因。
感谢回复, 这是在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,你强行映射为两个,这是你的这个错误的原因。