gpt 2 simple:Python软件包可以轻松地在新文本上重新训练OpenAI的GPT 2文本生成模型 源码

gpt-2-简单 一个简单的Python程序包,用于包装的(特别是“小” 124M和“中” 355M超参数版本)的现有模型微调和生成脚本。 此外,此程序包允许更容易地生成文本,生成到文件以方便管理,并允许使用前缀来强制文本以给定短语开头。 该软件包合并了以下内容并对其进行了最少的低级更改: 来自OpenAI的存储库的模型管理(MIT许可证) 从Neil Shepperd的GPT-2进行模型微调(MIT许可) 来自文本生成输出管理(MIT许可,也由我创建) 为了进行微调,强烈建议使用GPU,尽管您可以使用CPU生成(尽管速度要慢得多)。 如果您是在云端进行训练,则强烈建议使用带有映像