Question

我正在尝试保存微调的bert模型。我已经正确运行了代码-可以正常工作，并且在ipython控制台中，我可以调用getPrediction并使其产生结果。

我保存了体重文件（最高的是model.ckpt-333.data-00000-of-00001

我不知道如何保存模型以使其可重用。

我正在使用bert-tensorflow。

pd.DataFrame(A.toarray())

Answer 1

正如问题明确指出要保存模型一样，这是它的工作原理：

import torch
torch.save(model, 'path/to/model')

saved_model = torch.load('path/to/model')

Answer 2

我认为您可以将model.ckpt-333.data-00000-of-00001重命名为bert_model.ckpt，然后以与使用非微调模型相同的方式使用它。例如，运行

python run_classifier.py \
  --task_name=MRPC \
  --do_predict=true \
  --data_dir=$GLUE_DIR/MRPC \
  --vocab_file=$BERT_BASE_DIR/vocab.txt \
  --bert_config_file=$BERT_BASE_DIR/bert_config.json \
  --init_checkpoint=$TRAINED_CLASSIFIER

--init_checkpoint指向模型的目录，或运行bert-as-service

bert-serving-start -model_dir $TRAINED_CLASSIFIER

带有正确的-model_dir。

保存“微调”的伯特模型

2 个答案: