本文整理汇总了Python中pytorch_pretrained_bert.modeling.BertForPreTraining方法的典型用法代码示例。如果您正苦于以下问题：Python modeling.BertForPreTraining方法的具体用法？Python modeling.BertForPreTraining怎么用？Python modeling.BertForPreTraining使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类pytorch_pretrained_bert.modeling的用法示例。
在下文中一共展示了modeling.BertForPreTraining方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: convert_tf_checkpoint_to_pytorch
# 需要导入模块: from pytorch_pretrained_bert import modeling [as 别名]
# 或者: from pytorch_pretrained_bert.modeling import BertForPreTraining [as 别名]
def convert_tf_checkpoint_to_pytorch(tf_checkpoint_path, bert_config_file, pytorch_dump_path):
# Initialise PyTorch model
config = BertConfig.from_json_file(bert_config_file)
print("Building PyTorch model from configuration: {}".format(str(config)))
model = BertForPreTraining(config)
# Load weights from tf checkpoint
load_tf_weights_in_bert(model, tf_checkpoint_path)
# Save pytorch-model
print("Save PyTorch model to {}".format(pytorch_dump_path))
torch.save(model.state_dict(), pytorch_dump_path) 开发者ID:649453932，项目名称:Bert-Chinese-Text-Classification-Pytorch，代码行数:14，代码来源:convert_tf_checkpoint_to_pytorch.py