微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

GPT2 GPT2 训练实现

程序名称:GPT2

授权协议: MIT

操作系统: 跨平台

开发语言: Python

GPT2 介绍

非官方 GPT-2 训练实现,支持 GPU 和 TPU。

GPT-2 是一种基于 transformer 的大型语言模型,具有 15 亿个参数,在 800 万网页数据集上进行训练。

依赖:

GPU

pip3 install tensorflow-gpu regex

TPU

pip3 install tensorflow regex google-api-python-client oauth2client

下载模型

pip3 install requests tqdm

生成数据集

pip3 install ftfy tqdm newspaper3k

GPT2 官网

https://github.com/ConnorJL/GPT2

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐