KevinGeertjens
/

bert-classification-model

Feature Extraction

bert-classification

Model card Files Files and versions Community

KevinGeertjens commited on Dec 1, 2022

Commit

95e2d7e

•

1 Parent(s): dcb4bf1

Update bert_classification_model.py

Files changed (1) hide show

bert_classification_model.py +0 -2

bert_classification_model.py CHANGED Viewed

@@ -6,8 +6,6 @@ from torch import nn
 class BertClassificationModel(PreTrainedModel):
-    config_class = BertClassificationConfig
     def __init__(self, config, num_main_segment=None, num_sub_segment=None):
         super(BertClassificationModel, self).__init__(config=config)
         self.num_main_segment = num_main_segment if num_main_segment else config.num_main_segment

 class BertClassificationModel(PreTrainedModel):
     def __init__(self, config, num_main_segment=None, num_sub_segment=None):
         super(BertClassificationModel, self).__init__(config=config)
         self.num_main_segment = num_main_segment if num_main_segment else config.num_main_segment