adapt 8GB GPU configuration

3 years ago · d3fbbf7282
parent 28fa95b09b
commit d3fbbf7282
1 changed files with 33 additions and 0 deletions
--- a/qurator/sbb_ner/webapp/config-8GB-GPU.json
+++ 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
+    }
+  ]
+}