/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

5 votes204 views3 answers
Print command python
Python
0 vote189 views1 answer
Tuple: Create tuple
Python
24 votes155 views4 answers
Return max integer from list
Python
0 vote214 views1 answer
Tensorflow create custom initializer
Python
0 vote197 views1 answer
Python scramble up elements in list
Python
0 vote203 views1 answer
How to add legend on side of the chart python
Python
0 vote206 views1 answer
Dictionary convert 2 lists into a dictionary, use zip()
Python
7 votes196 views2 answers
Python string to datetime python
Python
31 votes218 views3 answers
Python array to list
Python
5 votes236 views3 answers
How to hide command console python
Python