📜  将 tensorflow 检查点转换为 pytorch - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:44.978000             🧑  作者: Mango

代码示例1
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