mirostat:纸的代码-“ Mirostat

时间:2021-05-26 14:05:25
【文件属性】:
文件名称:mirostat:纸的代码-“ Mirostat
文件大小:2KB
文件格式:ZIP
更新时间:2021-05-26 14:05:25
Python 在ICLR 2021中我们的论文“ Mirostat:一种困惑控制的神经文本解码算法”中提出的拟微节制采样算法的代码。可在以下位置获得该论文: ://arxiv.org/abs/2007.14966。 Tl; dr :我们提供了一种新的文本解码算法,该算法可直接控制生成的文本统计信息,并因此使用大型语言模型(例如GPT-2,CTRL等)生成更多类似于人类的文本。 安装要求: 点安装变压器== v2.8.0 使用示例: python mirostat.py --num_tokens 200 --tau 3.0 --context "/context.txt" 其中num_tokens代表要生成的令牌数量,tau代表平均意外值(即困惑日志),context.txt是包含上下文的文本文件。 如果您发现该代码在您的工作中很有用,请引用为: @inproceedings{ Bas
【文件预览】:
mirostat-master
----mirostat.py(3KB)
----README.md(1KB)

网友评论