o

araberta_setting/modeling_bilstm_crf.py

@@ -6,7 +6,8 @@ class BERT_BiLSTM_CRF(nn.Module):
     def __init__(self, base_model, config, dropout_rate=0.2, rnn_dim=256):
         super().__init__()
         self.bert = base_model
-        self.label2id = config.label2id  # <-- pulled from config
+        self.config = config              # ✅ add this line
+        self.label2id = config.label2id
         self.id2label = config.id2label
         self.num_labels = config.num_labels
 
@@ -16,7 +17,7 @@ class BERT_BiLSTM_CRF(nn.Module):
             num_layers=2,
             batch_first=True,
             bidirectional=True,
-            dropout=0.2
+            dropout=dropout_rate
         )
         self.dropout = nn.Dropout(dropout_rate)
         self.classifier = nn.Linear(rnn_dim * 2, self.num_labels)