本文整理汇总了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