/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

47 votes158 views3 answers
String to array python
Python
0 vote182 views2 answers
Np.exp in python numpy
Python
0 vote181 views1 answer
How to insert value in admin panel in django
Python
0 vote218 views1 answer
Django strptime
Python Django
0 vote191 views1 answer
Np.zeros data type not understood
Python
3 votes203 views2 answers
Python script that executes at time
Python
0 vote125 views1 answer
Fibonacci sequence in python
Python
0 vote173 views1 answer
Calculate time python
Python
2 votes162 views1 answer
Torch device
Python
0 vote186 views1 answer
Multiprocessing print does not work
Python