BERT训练检查站

Question

我通过BERT训练了数据，并在bert_output目录中获得了以下文件：

• model.ckpt-0.data-00000-of-00001
• model.ckpt-0.index
• model.ckpt-0.meta
• model.ckpt-343.data-00000-of-00001
• model.ckpt-343.index
• model.ckpt-343.meta
• 检查点
• graph.pbtxt

现在，我需要确定最高编号的model.ckptfile的编号。这组文件包含我训练的模型的权重。

我的问题是：如何读取.ckpt文件中的内容？

---

export BERT_BASE_DIR=./path/to/weights/downloaded/in/step2
export TRAINED_CLASSIFIER=./bert_output/model.ckpt-[highest checkpoint number you saw]
python bert/run_classifier.py \
--task_name=cola \
--do_predict=true \
--data_dir=./data \
--vocab_file=$BERT_BASE_DIR/vocab.txt \
--bert_config_file=$BERT_BASE_DIR/bert_config.json \
--init_checkpoint=$TRAINED_CLASSIFIER \
--max_seq_length=128 \
--output_dir=./bert_output/