GPT的实现细节

关于GPT的代码细节，这里梳理了一下：

数据集构造

原始数据集schema：

input=who is your favorite basketball player?
output=Of course Kobe Bryant!

那么在构造训练集时，根据chunk size构造多个输入：

input_1=who is your favorite basketball player? Of
input_2=who is your favorite basketball player? Of course
......
input_n-1=who is your favorite basketball player? Of course Kobe Bryant!
input_n=who is your favorite basketball player? Of course Kobe Bryant! <EOS>

由于训练任务是下一个单词预测，所以 $x=input[:-1],y=input[1:]$

loss

$x$是模型可见已知的，需要mask掉，不算入loss：

y[:-1] = -1
loss = F.cross_entropy(logits.view(-1, logits.size(-1)), targets.view(-1), ignore_index=-1)

生成

在karpathy/minGPT项目中，是直接粗暴地生成固定长度的文本。这样做的问题就是生成的文本无法判断何处阶段。

在构造模型输入的时候，我们就加入了 <EOS> token，来标记文本的结束。那么在推理阶段，如果碰到该token，则结束生成：

if token == "<EOS>":break

参考

• GPT2LMHeadModel
• mingpt
• GPT2-NewsTitle