File size: 308 Bytes
526f371
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from transformers import PretrainedConfig


class OnsidesConfig(PretrainedConfig):
    model_type = "onsides"

    def __init__(self, classifier_dropout=0.5, num_labels=2, **kwargs):
        self.classifier_dropout = classifier_dropout
        self.num_labels = num_labels
        super().__init__(**kwargs)