| 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) | |