Spaces:
Running
Running
Commit
·
8aad65b
1
Parent(s):
f2b6c86
Исправлен путь к модели
Browse files
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.
|
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):
|