KevinGeertjens
commited on
Commit
•
95e2d7e
1
Parent(s):
dcb4bf1
Update bert_classification_model.py
Browse files
bert_classification_model.py
CHANGED
@@ -6,8 +6,6 @@ from torch import nn
|
|
6 |
|
7 |
|
8 |
class BertClassificationModel(PreTrainedModel):
|
9 |
-
config_class = BertClassificationConfig
|
10 |
-
|
11 |
def __init__(self, config, num_main_segment=None, num_sub_segment=None):
|
12 |
super(BertClassificationModel, self).__init__(config=config)
|
13 |
self.num_main_segment = num_main_segment if num_main_segment else config.num_main_segment
|
|
|
6 |
|
7 |
|
8 |
class BertClassificationModel(PreTrainedModel):
|
|
|
|
|
9 |
def __init__(self, config, num_main_segment=None, num_sub_segment=None):
|
10 |
super(BertClassificationModel, self).__init__(config=config)
|
11 |
self.num_main_segment = num_main_segment if num_main_segment else config.num_main_segment
|