ckpt to torch

Convert the tensorflow version of the pre-trained BERT model to the pytorch version of the pre-trained model

https://huggingface.co/transformers/converting_tensorflow_models.html

export BERT_BASE_DIR=/path/to/bert/uncased_L-12_H-768_A-12

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

Guess you like

Origin blog.csdn.net/u011939633/article/details/108710186