/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 vote138 views1 answer
Trim string to max length python
Python
1 vote118 views1 answer
Find substring in string python
Python
15 votes229 views6 answers
Matplotlib.pyplot
Python
0 vote147 views1 answer
How to create superuser in django heroku
Python
0 vote151 views1 answer
Python 3.7 install snakemake
Python
0 vote158 views1 answer
Ball python
Python
0 vote146 views1 answer
How to Set Axis Range (xlim, ylim) in Matplotlib
Python
0 vote133 views1 answer
How to copy content of one file to another in python
Python
1 vote157 views1 answer
Append to list in dictionary python if exists
Python
75 votes140 views3 answers
How to get a random element from an array in python
Python