cs224n_2019/pytorch-bert-code/bert/README.md

how to convert bert Converting Tensorflow Checkpoints to pytorch model file

如何将bert model 的Tensorflow模型 转换为pytorch模型

下载Tensorflow模型文件 解压缩到文件夹下

应该有

• bert_config.json

• bert_model.ckpt.data-00000-of-00001

• bert_model.ckpt.index

• bert_model.ckpt.meta

• vocab.txt

这几个文件 运行run.sh 后生成对应pytorch_model.bin

具体代码

export BERT_BASE_DIR=。/

transformers bert \
  $BERT_BASE_DIR/bert_model.ckpt \
  $BERT_BASE_DIR/bert_config.json \
  $BERT_BASE_DIR/pytorch_model.bin

原来convert_tf_checkpoint_to_pytorch.py被新版本废除