Update config.json

config.json

@@ -1,18 +1,21 @@
-{
-    "architectures": [
-      "FlairNER"
-    ],
-    "embedding_dim": 300,
-    "hidden_size": 256,
-    "dropout": 0.5,
-    "num_classes": 19,
-    "rnn_layers": 1,
-    "bidirectional": true,
-    "crf": true,
-    "word_dropout": 0.05,
-    "locked_dropout": 0.5,
-    "batch_size": 64,
-    "learning_rate": 0.1,
-    "max_epochs": 100
-  }
-  
+{
+  "model_type": "flair",
+  "architectures": [
+    "FlairNER"
+  ],
+  "embedding_dim": 300,
+  "hidden_size": 256,
+  "dropout": 0.5,
+  "num_classes": 19,
+  "rnn_layers": 1,
+  "bidirectional": true,
+  "crf": true,
+  "word_dropout": 0.05,
+  "locked_dropout": 0.5,
+  "linear_in_features": 512,
+  "linear_out_features": 19,
+  "loss_function": "ViterbiLoss",
+  "learning_rate": 0.1,
+  "batch_size": 64,
+  "max_epochs": 100
+}