/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 vote299 views1 answer
Module installed but not found python
Python
0 vote313 views1 answer
Python sum lists element wise
Python
0 vote303 views1 answer
Poision in chinese
Python
0 vote345 views1 answer
Selenium python tkinter
Python
3 votes335 views1 answer
Json not readable python
Python
1 vote303 views1 answer
How to find what is the response from the server with python
Python
-1 vote339 views1 answer
Transpose 3d matrix pytorch
Python
2 votes314 views1 answer
Replacing spaces in a string python
Python
3 votes314 views1 answer
Corr pandas
Python
0 vote355 views1 answer
ID number zero python
Python