mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 21:00:08 +06:00
Merge branch 'master' of https://github.com/huggingface/pytorch-pretrained-BERT
This commit is contained in:
commit
eaa6db92f1
@ -487,7 +487,7 @@ class BertForQuestionAnswering(nn.Module):
|
||||
m.weight.data.normal_(config.initializer_range)
|
||||
elif isinstance(m, BERTLayerNorm):
|
||||
m.beta.data.normal_(config.initializer_range)
|
||||
m.gamme.data.normal_(config.initializer_range)
|
||||
m.gamma.data.normal_(config.initializer_range)
|
||||
if isinstance(m, nn.Linear):
|
||||
m.bias.data.zero_()
|
||||
self.apply(init_weights)
|
||||
|
Loading…
Reference in New Issue
Block a user