from transformers import PretrainedConfig


class MRIBrainSequenceBERTConfig(PretrainedConfig):
    model_type = "mri_brain_sequence_bert"

    def __init__(self, max_len=512, dropout=0.2, num_classes=17, **kwargs):
        self.max_len = max_len
        self.dropout = dropout
        self.num_classes = num_classes
        super().__init__(**kwargs)