文件名称：mirostat:纸的代码-“ Mirostat
文件大小：2KB
文件格式：ZIP
更新时间：2024-06-15 07:52:05
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)