K-BERT icon indicating copy to clipboard operation
K-BERT copied to clipboard

pytorch 转 tensorflow

Open zxy951005 opened this issue 3 years ago • 1 comments

autoliuweijie您好: 这个模型可以转为tensorflow格式或者有tensorflow版吗?主要是对比实验需要用您这个模型。 谢谢!

zxy951005 avatar Dec 10 '20 02:12 zxy951005

autoliuweijie您好: 这个模型可以转为tensorflow格式或者有tensorflow版吗?主要是对比实验需要用您这个模型。 谢谢!

目前没提供转tensorflow的脚本,但是你可以根据模型参数名称,自己写一个转换脚本来实现。

参考UER里的转换脚本:https://github.com/dbiir/UER-py/blob/master/scripts/convert_bert_from_uer_to_google.py

autoliuweijie avatar Dec 10 '20 12:12 autoliuweijie