File size: 352 Bytes
0a015a7 fa7dbe5 0a015a7 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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)
|