comment on gelu function
This commit is contained in:
1 parent
01b1a05343
commit
e6d106a01b
1 file changed
+2
-2
+2
-2
@@ -28,8 +28,8 @@ from torch.nn import CrossEntropyLoss
|
||||
|
||||
def gelu(x):
|
||||
return x * 0.5 * (1.0 + torch.erf(x / math.sqrt(2.0)))
|
||||
# OpenAI GPT gelu version :
|
||||
# return 0.5 * x * (1 + torch.tanh(math.sqrt(2 / math.pi) * (x + 0.044715 * torch.pow(x, 3))))
|
||||
# For information: OpenAI GPT gelu version is a bit different:
|
||||
# 0.5 * x * (1 + torch.tanh(math.sqrt(2 / math.pi) * (x + 0.044715 * torch.pow(x, 3))))
|
||||
|
||||
|
||||
class BertConfig(object):
|
||||
|
||||
Reference in new issue
Block a user