codertimo / codertimo/BERT-pytorch

transformer.py 中的forword方法调用的SublayerConnection类。实现残差链接和标准化的实现

Open
#83 2 comments 3 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
6.5k
Forks
1.3k
PR merge metrics
No merged PRs in 30d

Description

sublayerout = layerNorm(x +sublayer(x))
**首先是残差链接然后是层标准化**
在你代码中:sublayer.py中 应该是
def forward(self, x, sublayer):
"Apply residual connection to any sublayer with the same size."
# return x + self.dropout(sublayer(self.norm(x)))
return self.norm( x + self.dropout(sublayer(x)))

tranformer.py中:
def forward(self, x, mask):
x = self.input_sublayer(x, lambda _x: self.attention.forward(_x, _x, _x, mask=mask))
x = self.output_sublayer(x, **lambda _x: self.feed_forward.forward(_x))**
return self.dropout(x)

此处我对论文立即额和你不一样,有错误的地方请指教

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.