omarmomen
/

sf_babylm_1

Model card Files Files and versions Community

omarmomen commited on Feb 3

Commit

3ca263d

•

1 Parent(s): 841b081

Update structformer.py

Files changed (1) hide show

structformer.py +1 -0

structformer.py CHANGED Viewed

@@ -808,6 +808,7 @@ class StructFormerModelForSequenceClassification(PreTrainedModel):
               weight_act=config.weight_act
         )
         self.config = config
         self.model.classifier = ClassificationHead(config)
     def _init_weights(self, module):

               weight_act=config.weight_act
         )
         self.config = config
+        self.num_labels = config.num_labels
         self.model.classifier = ClassificationHead(config)
     def _init_weights(self, module):