/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 vote179 views2 answers
Check if set is a subset of another python
Python
2 votes172 views1 answer
Fyit download
Python
0 vote206 views1 answer
Qtablewidget not editable python
Python
0 vote214 views1 answer
How to get parent model object based on child model filter in django
Python Django
1 vote195 views1 answer
Panda dataframe read csv change string to float
Python
0 vote167 views1 answer
Read to file in python
Python
0 vote175 views4 answers
Insert blank row in data frame
Python
8 votes245 views3 answers
How to get index in python
Python
1 vote196 views1 answer
Python filter None dictionary
Python
8 votes187 views1 answer
Delete certain characters from a string python
Python