mirror of
https://github.com/qurator-spk/sbb_ner.git
synced 2025-06-09 04:10:08 +02:00
adapt 8GB GPU configuration
This commit is contained in:
parent
28fa95b09b
commit
d3fbbf7282
1 changed files with 33 additions and 0 deletions
33
qurator/sbb_ner/webapp/config-8GB-GPU.json
Normal file
33
qurator/sbb_ner/webapp/config-8GB-GPU.json
Normal 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
|
||||
}
|
||||
]
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue