Update modelling_RW.py
Browse files- modelling_RW.py +1 -1
modelling_RW.py
CHANGED
@@ -959,7 +959,7 @@ class RWForTokenClassification(RWPreTrainedModel):
|
|
959 |
else:
|
960 |
classifier_dropout = 0.1
|
961 |
self.dropout = nn.Dropout(classifier_dropout)
|
962 |
-
self.classifier = nn.Linear(config.hidden_size, config.
|
963 |
|
964 |
# Initialize weights and apply final processing
|
965 |
self.post_init()
|
|
|
959 |
else:
|
960 |
classifier_dropout = 0.1
|
961 |
self.dropout = nn.Dropout(classifier_dropout)
|
962 |
+
self.classifier = nn.Linear(config.hidden_size, config.num_labels)
|
963 |
|
964 |
# Initialize weights and apply final processing
|
965 |
self.post_init()
|