adapt 8GB GPU configuration

pull/2/head
Kai 3 years ago
parent 28fa95b09b
commit d3fbbf7282

@ -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…
Cancel
Save