johnny12150 / gc-san Goto Github PK

Use class MultiHeadedAttention not nn.MultiheadAttention, how to solve it? thx

作者您好，您的注意力机制在使用时很方便，但是多头注意力怎么运行不起来，是这样就可以吗
self.multihead_attn = MultiHeadedAttention(4, self.hidden_size, 0.2).cuda()

Data in sample_train-item-views.csv use session_id& item_id as the column name . But the code in line 56 & 57 is like data['session_id'] = data.pop('sessionId') .This will lead to error when testing sample data. So the code should be modified as data['session_id'] = data.pop('session_id') and data['item_id'] = data.pop('item_id'). The error disappeared and the code produces sample txt file normally after this modification👍