autoliuweijie / K-BERT

Source code of K-BERT (AAAI2020)
https://ojs.aaai.org//index.php/AAAI/article/view/5681
949 stars 212 forks source link

pytorch 转 tensorflow #49

Open zxy951005 opened 3 years ago

zxy951005 commented 3 years ago

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

autoliuweijie commented 3 years ago

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

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

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