Upload modeling_distil_greek_news_bert.py

modeling_distil_greek_news_bert.py

@@ -13,7 +13,7 @@ class DistilGreekNewsBert(DistilBertPreTrainedModel):
 
         n_cls = config.num_labels_class
         n_ner = config.num_labels_ner
-        self.ner_loss_weight = getattr(config, ner_loss_weight, 3.0)
+        self.ner_loss_weight = getattr(config, "ner_loss_weight", 3.0)
 
         self.class_dropout = nn.Dropout(0.3)
         self.class_fc   = nn.Linear(config.dim, 768)