/Python/Convert tensorflow checkpoint to pytorch

Convert tensorflow checkpoint to pytorch

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

transformers-cli convert --model_type bert \
  --tf_checkpoint $BERT_BASE_DIR/bert_model.ckpt \
  --config $BERT_BASE_DIR/bert_config.json \
  --pytorch_dump_output $BERT_BASE_DIR/pytorch_model.bin

Source: huggingface.co

Similar codes

0 vote358 views1 answer
Numpy add
Python
1 vote341 views1 answer
Check how many times a substring appears in a string
Python
0 vote385 views2 answers
How to get current page url in django template
Python Django
62 votes335 views6 answers
Sort dictionary by values
Python
0 vote381 views1 answer
Flask windows auto reload
Python Flask
4 votes325 views1 answer
Python windows hide files
Python
1 vote347 views1 answer
Django return only part of string
Python Django
1 vote310 views1 answer
Breaking big csv into chunks pandas
Python
2 votes312 views1 answer
Python search for word is in column
Python
0 vote330 views1 answer
Gnome-shell turn off
Python