从头开始训练拥抱的GPT2：断言n_state％config.n_head == 0错误

Question

我正在尝试将GPT2架构用于音乐应用，因此需要从头开始进行培训。经过一番谷歌搜索后，我发现拥抱面的github中的问题＃1714已经“解决”了这个问题。当我尝试运行提议的解决方案时：

from transformers import GPT2Config, GPT2Model

NUMLAYER = 4
NUMHEAD = 4
SIZEREDUCTION = 10 #the factor by which we reduce the size of the velocity argument.
VELSIZE = int(np.floor(127/SIZEREDUCTION)) + 1 
SEQLEN=40 #size of data sequences.
EMBEDSIZE = 5 

config = GPT2Config(vocab_size = VELSIZE, n_positions = SEQLEN, n_embd = EMBEDSIZE, n_layer = NUMLAYER, n_ctx = SEQLEN, n_head = NUMHEAD)  
model = GPT2Model(config)

我收到以下错误：

Traceback (most recent call last):

  File "<ipython-input-7-b043a7a2425f>", line 1, in <module>
    runfile('C:/Users/cnelias/Desktop/PHD/Swing project/code/script/GPT2.py', wdir='C:/Users/cnelias/Desktop/PHD/Swing project/code/script')

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 786, in runfile
    execfile(filename, namespace)

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "C:/Users/cnelias/Desktop/PHD/Swing project/code/script/GPT2.py", line 191, in <module>
    model = GPT2Model(config)

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\transformers\modeling_gpt2.py", line 355, in __init__
    self.h = nn.ModuleList([Block(config.n_ctx, config, scale=True) for _ in range(config.n_layer)])

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\transformers\modeling_gpt2.py", line 355, in <listcomp>
    self.h = nn.ModuleList([Block(config.n_ctx, config, scale=True) for _ in range(config.n_layer)])

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\transformers\modeling_gpt2.py", line 223, in __init__
    self.attn = Attention(nx, n_ctx, config, scale)

  File "C:\Users\cnelias\Anaconda3\lib\site-packages\transformers\modeling_gpt2.py", line 109, in __init__
    assert n_state % config.n_head == 0

这是什么意思，我该如何解决？

[更笼统地说，是否有关于如何使用GPT2进行前向呼叫的文档？我可以定义自己的train()函数还是必须使用模型的内置函数？我是否被迫使用Dataset进行训练，或者可以给它单独张量？我在寻找它，但是在文档中找不到这些答案，但是也许我错过了一些东西。

PS：我已经阅读了fron拥抱face.co的博客文章，但是它忽略了太多的信息和细节，无法对我的应用程序有用。

Answer 1

我认为错误消息很清楚：

assert n_state % config.n_head == 0

通过the code追溯，我们可以看到

n_state = nx # in Attention: n_state=768

表示n_state表示嵌入尺寸（在类似BERT的模型中默认情况下通常为768）。然后，当我们查看GPT-2 documentation时，似乎指定该参数的参数为​​n_embd，您将其设置为5。如错误所示，嵌入尺寸必须通过注意头的数量均匀地整除，其指定为4。因此，选择不同的嵌入尺寸作为4的倍数即可解决该问题。当然，您也可以更改开头的头数，但似乎不支持奇数的嵌入尺寸。