From d3fbbf7282be20c9548d35680b6080c12df209e0 Mon Sep 17 00:00:00 2001 From: Kai Date: Mon, 9 Aug 2021 16:46:06 +0200 Subject: [PATCH] adapt 8GB GPU configuration --- qurator/sbb_ner/webapp/config-8GB-GPU.json | 33 ++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 qurator/sbb_ner/webapp/config-8GB-GPU.json 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