皮托伯特
使用PyTorch TransformerEncoder和预训练模型的BERT实现。
安装
pip install pytorch-bert
用法
config = BertConfig . from_json ( "path-to-pretarined-weights/bert_config.json" )
model = Bert ( config )
load_tf_weight_to_pytorch_bert ( model , config , "path-to-pretarined-weights/bert_model.ckpt" )
将模型文件下载到存储库中。
1