/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

44 votes120 views6 answers
Json python
Python
0 vote121 views1 answer
Python pie chart with legend
Python
-2 votes101 views2 answers
Django form date picker
Python
0 vote120 views1 answer
Key item loop list python
Python
0 vote147 views1 answer
Create a django and react readonly web app
Python Django
10 votes100 views6 answers
Range function in python
Python
-4 votes104 views5 answers
Current django version
Python
0 vote118 views2 answers
Cumulative chart python plotly
Python
0 vote114 views1 answer
Python remove suffix
Python
1 vote104 views1 answer
Python find string in list
Python