convert tensorflow checkpoint to pytorch

Solutions on MaxInterview for convert tensorflow checkpoint to pytorch by the best coders in the world

we are a community of more than 2 million smartest coders
registration for
employee referral programs
are now open
get referred to google, amazon, flipkart and more
register now
  
pinned-register now
showing results for - "convert tensorflow checkpoint to pytorch"
Alonso
24 Jan 2018
1export BERT_BASE_DIR=/path/to/bert/uncased_L-12_H-768_A-12
2
3transformers-cli convert --model_type bert \
4  --tf_checkpoint $BERT_BASE_DIR/bert_model.ckpt \
5  --config $BERT_BASE_DIR/bert_config.json \
6  --pytorch_dump_output $BERT_BASE_DIR/pytorch_model.bin
7