/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 vote96 views1 answer
Square finder python
Python
-1 vote100 views1 answer
Python : search file for string
Python
0 vote96 views1 answer
Liste compréhension python
Python
0 vote109 views1 answer
Textrank python implementation
Python
1 vote100 views1 answer
Increase contrast cv2
Python
0 vote98 views1 answer
No module named 'filterpy'
Python
-1 vote102 views1 answer
Binascii.Error: Incorrect padding in python django
Python
2 votes130 views1 answer
Matplotlib savefig not working
Python
0 vote106 views1 answer
Items of a list not in another list python
Python
2 votes96 views1 answer
Python how to unnest a nested list
Python