在Bert的预训练模型中,主流的模型都是以tensorflow的形势开源的。但是huggingface在Transformers中提供了一份可以转换的接口(convert_bert_original_tf_checkpoint_to_pytorch.py)。
但是如何在windows的IDE中执行呢?
pip install transformers
import transformers.models.bert.convert_bert_original_tf_checkpoint_to_pytorch as con con.convert_tf_checkpoint_to_pytorch( r'.\bert_model.ckpt', r'.\bert_config.json', r'.\pytorch_bert.bin' )
convert_tf_checkpoint_to_pytorch中三个参数分别是:bert模型名称、config文件地址,输出的pytorch文件保存地址