跳到正文
原文
Hugging Face Blog·· 2020-03-01精选AI 评分62

如何用 Transformers 生成文本:不同解码方法详解

How to generate text: using different decoding methods for language generation with Transformers

AI 导读

Hugging Face 发布教程,介绍如何用 transformers 库实现自回归语言生成的多种解码方法,包括贪心搜索、beam search 和采样。文中以 GPT2 为例给出可运行代码,演示 num_beams、no_repeat_ngram_size、temperature、top_k、top_p 等参数的效果。

推荐理由

系统梳理贪心、beam search 与采样等解码策略的差异,并给出 transformers 中可直接运行的参数配置。

来源:Hugging Face Blog · huggingface.co