tf-to-torch.sh 332 Bytes
Newer Older
hepj987's avatar
hepj987 committed
1
2
3
4
5
python3 convert_tf_checkpoint.py \
 --bert_model "bert-large-uncased" \
 --tf_checkpoint  /public/home/hepj/model_source/uncased_L-24_H-1024_A-16/bert_model.ckpt \
 --bert_config_path /public/home/hepj/model_source/uncased_L-24_H-1024_A-16/bert_config.json \
 --output_checkpoint /public/home/hepj/model_source/model_pytorch.ckpt.pt