Bert4Rec代码报错Invalid argument: Key: XXX. Can‘t parse serialized Example.

运行Bert4Rec代码时,先使用gen_data.py分割数据,后运行run.py进行训练,但是训练时报错Invalid argument: Key: XXX. Can‘t parse serialized Example.

找了很久发现了问题产生的原因:
gen_data.py和run.py中设置的max_seq_length参数值不一样,将值设置为一样的就可以了。

参考:
https://github.com/GanjinZero/ChineseEHRBert/issues/10

你可能感兴趣的:(tensorflow深度学习,python,bert,tensorflow)