qurator-spk/sbb_ner
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_ner.git synced 2025-06-09 12:20:00 +02:00
Code Issues Releases Wiki Activity

adapt 8GB GPU configuration

Browse source
This commit is contained in:
Kai 2021-08-09 16:46:06 +02:00
parent 28fa95b09b
commit d3fbbf7282
1 changed files with 33 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

33
qurator/sbb_ner/webapp/config-8GB-GPU.json Normal file
View file

@ -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
}
]
}