diff --git a/qurator/sbb_ner/webapp/config-8GB-GPU.json b/qurator/sbb_ner/webapp/config-8GB-GPU.json new file mode 100644 index 0000000..b1e1c14 --- /dev/null +++ b/qurator/sbb_ner/webapp/config-8GB-GPU.json @@ -0,0 +1,33 @@ +{ + "BATCH_SIZE": 16, + "MODELS": [ + { + "name": "DC-SBB + CONLL + GERMEVAL", + "id": 1, + "model_dir": "data/konvens2019/build-wd_0.03/bert-all-german-de-finetuned", + "epoch": 7, + "default": true + }, + { + "name": "DC-SBB + CONLL + GERMEVAL + SBB", + "id": 2, + "model_dir": "data/konvens2019/build-on-all-german-de-finetuned/bert-sbb-de-finetuned", + "epoch": 7, + "default": false + }, + { + "name": "DC-SBB + SBB", + "id": 3, + "model_dir": "data/konvens2019/build-wd_0.03/bert-sbb-de-finetuned", + "epoch": 7, + "default": false + }, + { + "name": "CONLL + GERMEVAL", + "id": 4, + "model_dir": "data/konvens2019/build-wd_0.03/bert-all-german-baseline", + "epoch": 7, + "default": false + } + ] +} \ No newline at end of file