MikhailPugachev commited on
Commit
8aad65b
·
1 Parent(s): f2b6c86

Исправлен путь к модели

Browse files
Files changed (1) hide show
  1. model_SingleLabelClassifier.py +1 -1
model_SingleLabelClassifier.py CHANGED
@@ -15,7 +15,7 @@ class SingleLabelClassifier(nn.Module):
15
  self.intermediate = nn.Linear(self.base.config.hidden_size, hidden_size)
16
  self.norm = nn.LayerNorm(hidden_size)
17
  self.activation = nn.ReLU()
18
- self.dropout = nn.Dropout(0.4)
19
  self.classifier = nn.Linear(hidden_size, num_labels)
20
 
21
  def forward(self, input_ids, attention_mask=None, token_type_ids=None,labels=None):
 
15
  self.intermediate = nn.Linear(self.base.config.hidden_size, hidden_size)
16
  self.norm = nn.LayerNorm(hidden_size)
17
  self.activation = nn.ReLU()
18
+ self.dropout = nn.Dropout(0.5)
19
  self.classifier = nn.Linear(hidden_size, num_labels)
20
 
21
  def forward(self, input_ids, attention_mask=None, token_type_ids=None,labels=None):