将bert预训练模型转换为PyTorch版本

1、GitHub - google-research/bert: TensorFlow code and pre-trained models for BERT

将bert预训练模型转换为PyTorch版本_第1张图片

2、下载下来,解压得到带有五个文件的文件夹。

3、打开命令行窗口

输入以下命令(这边的地址自己选择):

python convert_tf_checkpoint_to_pytorch.py --tf_checkpoint_path D:\自然语言工具包\chinese_L-12_H-768_A-12\bert_model.ckpt --bert_config_file D:\自然语言工具包\chinese_L-12_H-768_A-12\bert_config.json --pytorch_dump_path D:\自然语言工具包\chinese_L-12_H-768_A-12\pytorch_model.bin

输入上面的命令就可以生成下面的bin文件了

5、最后,将文件夹中的bert_config.json文件复制一份,重新命名为config.json,这样就可以啦!!!

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