diff --git a/model/config_training_bert.jsonnet b/model/config_training_bert.jsonnet
index 824dabd3acbbfe53621736fdd4178cef28fb15fd..b65c0af8be72fa4a7a63145449618090d830761a 100644
--- a/model/config_training_bert.jsonnet
+++ b/model/config_training_bert.jsonnet
@@ -22,6 +22,8 @@
     },
     "model": {
         "type": "simple_tagger",
+        "calculate_span_f1": true,
+        "label_encoding": "BIOUL",
         "encoder": {
             "type": "lstm",
             "bidirectional": true,
@@ -67,13 +69,14 @@
     "train_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_train.ner.conllu",
     "validation_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_dev.ner.conllu",
     "trainer": {
-        "cuda_device": 1,
+        "cuda_device": -1,
         "grad_norm": 5,
         "num_epochs": 4,
-        "num_serialized_models_to_keep": 3,
+        "num_serialized_models_to_keep": 1,
         "optimizer": {
             "type": "bert_adam",
             "lr": 0.001
-        }
+        },
+        "validation_metric": "+f1-measure-overall"
     }
 }
\ No newline at end of file