Update structformer.py
Browse files- structformer.py +1 -0
structformer.py
CHANGED
|
@@ -808,6 +808,7 @@ class StructFormerModelForSequenceClassification(PreTrainedModel):
|
|
| 808 |
weight_act=config.weight_act
|
| 809 |
)
|
| 810 |
self.config = config
|
|
|
|
| 811 |
self.model.classifier = ClassificationHead(config)
|
| 812 |
|
| 813 |
def _init_weights(self, module):
|
|
|
|
| 808 |
weight_act=config.weight_act
|
| 809 |
)
|
| 810 |
self.config = config
|
| 811 |
+
self.num_labels = config.num_labels
|
| 812 |
self.model.classifier = ClassificationHead(config)
|
| 813 |
|
| 814 |
def _init_weights(self, module):
|