
Albert-PyTorch:一个高效的、精简的证书-源码,用于实现自我监督学习中的语言表示。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
该存储库提供 Albert 模型在 PyTorch 中的实现,位于中兰。 依赖于 PyTorch 版本 1.10,CUDA 版本 9.0,以及 cuDNN 版本 7.5,并利用 scikit-learn 进行句子级别的处理。 为了方便使用,提供了预训练的英语模型官方下载链接。 为了适应此版本的配置,建议下载相应的 PyTorch 模型(通过 Google 驱动器),包括 v1、v2 以及微调版本 1。 请将 `config.json` 文件和 `30k-clean.model` 文件复制到 `prev_trained_model/albert_base_v2` 目录中。 例如,目录结构如下:
├── prev_trained_model
│ └── albert_base_v2
│ ├── pytorch_model.bin
│ ├── config.json
│ └── 30k-clean.model
此外,还提供了将 Albert TF 检查点转换为 PyTorch 的 Python 脚本 `convert_albert_tf_checkpoint_to_pytorch.py`,用于转换 `. / prev_trained_m` 中的 TF 检查点文件。
全部评论 (0)
还没有任何评论哟~


