トーチへのckpt

事前トレーニング済みBERTモデルのテンソルフローバージョンを事前トレーニング済みモデルのpytorchバージョンに変換します

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

エクスポートBERT_BASE_DIR = / path / to / bert / uncased_L-12_H-768_A-12

トランスフォーマーbert $ BERT_BASE_DIR / bert_model.ckpt $ BERT_BASE_DIR / bert_config.json $ BERT_BASE_DIR / pytorch_model.bin

おすすめ

転載: blog.csdn.net/u011939633/article/details/108710186