File size: 560 Bytes
74a45d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import PretrainedConfig, BertConfig
class TemporalRelationClassificationConfig(BertConfig):
model_type = "TemporalRelationClassification"
def __init__(self, special_markers=False, pool_tokens=False, ES_ID=0, EMS1=0, EMS2=0, architecture=0, **kwargs):
super().__init__(**kwargs)
self.base_lm = self.name_or_path
self.pool_tokens = pool_tokens
self.special_markers = special_markers
self.architecture = architecture
self.EMS1 = EMS1
self.EMS2 = EMS2
self.ES_ID = ES_ID
|