souvik-nlp commited on
Commit
32199f2
·
verified ·
1 Parent(s): 5680a5a

Add configuration_dualmedbert.py for custom architecture

Browse files
Files changed (1) hide show
  1. configuration_dualmedbert.py +17 -0
configuration_dualmedbert.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PretrainedConfig
2
+
3
+ class DualMedBertConfig(PretrainedConfig):
4
+ model_type = "dualmedbert"
5
+ def __init__(
6
+ self,
7
+ num_classes=27,
8
+ lora_rank=8,
9
+ lora_alpha=32,
10
+ student_ck="distilbert-base-uncased",
11
+ **kwargs
12
+ ):
13
+ super().__init__(**kwargs)
14
+ self.num_classes = num_classes
15
+ self.lora_rank = lora_rank
16
+ self.lora_alpha = lora_alpha
17
+ self.student_ck = student_ck