文件名称:pytorch-bert:使用PyTorch的TransformerEncoder实现BERT
文件大小:18KB
文件格式:ZIP
更新时间:2024-05-15 15:41:03
nlp machine-learning python3 pytorch bert
皮托伯特 使用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" ) 将模型文件下载到存储库中。
【文件预览】:
pytorch-bert-master
----setup.py(801B)
----.gitignore(2KB)
----requirements.txt(6B)
----pyproject.toml(75B)
----setup.cfg(182B)
----examples()
--------run_squad.py(5KB)
----pytorch_bert()
--------__init__.py(292B)
--------feature.py(2KB)
--------tokenizer.py(10KB)
--------modeling.py(5KB)
--------weight_converter.py(5KB)
----README.md(1KB)
----tests()
--------test_tokenizer.py(3KB)
--------__init__.py(0B)
--------test_modeling.py(2KB)
--------test_feature.py(4KB)
--------test_converter.py(3KB)
----.circleci()
--------config.yml(1KB)
----.gitattributes(66B)
----requirements-dev.txt(104B)
----tox.ini(229B)