fixed init_weight

This commit is contained in:
erenup committed 2019-09-16 19:55:24 +08:00
1 parent 982f181aa7
commit a9debaca3d
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -418,7 +418,7 @@ class RobertaForMultipleChoice(BertPreTrainedModel):
self.dropout = nn.Dropout(config.hidden_dropout_prob)
self.classifier = nn.Linear(config.hidden_size, 1)
self.apply(self.init_weights)
self.init_weights()
def forward(self, input_ids, token_type_ids=None, attention_mask=None, labels=None,
position_ids=None, head_mask=None):