readme: add dataset information to metadata (#2)
Browse files- readme: add dataset information to metadata (6d2d097bf732fdde5274fdd47f5e3ce4dd32cba4)
README.md
CHANGED
@@ -9,6 +9,8 @@ tags:
|
|
9 |
- hetzner-gex44
|
10 |
- hetzner-gpu
|
11 |
base_model: dbmdz/bert-base-german-cased
|
|
|
|
|
12 |
widget:
|
13 |
- text: Wesentliche Tätigkeiten der Compliance-Funktion wurden an die Mercurtainment
|
14 |
AG , Düsseldorf , ausgelagert .
|
@@ -39,8 +41,8 @@ fine-tuning.
|
|
39 |
|
40 |
A hyper-parameter search over the following parameters with 5 different seeds per configuration is performed:
|
41 |
|
42 |
-
* Batch Sizes: [`
|
43 |
-
* Learning Rates: [`
|
44 |
|
45 |
More details can be found in this [repository](https://github.com/stefan-it/co-funer). All models are fine-tuned on a
|
46 |
[Hetzner GX44](https://www.hetzner.com/dedicated-rootserver/matrix-gpu/) with an NVIDIA RTX 4000.
|
|
|
9 |
- hetzner-gex44
|
10 |
- hetzner-gpu
|
11 |
base_model: dbmdz/bert-base-german-cased
|
12 |
+
datasets:
|
13 |
+
- stefan-it/co-funer
|
14 |
widget:
|
15 |
- text: Wesentliche Tätigkeiten der Compliance-Funktion wurden an die Mercurtainment
|
16 |
AG , Düsseldorf , ausgelagert .
|
|
|
41 |
|
42 |
A hyper-parameter search over the following parameters with 5 different seeds per configuration is performed:
|
43 |
|
44 |
+
* Batch Sizes: [`8`, `16`]
|
45 |
+
* Learning Rates: [`5e-05`, `3e-05`]
|
46 |
|
47 |
More details can be found in this [repository](https://github.com/stefan-it/co-funer). All models are fine-tuned on a
|
48 |
[Hetzner GX44](https://www.hetzner.com/dedicated-rootserver/matrix-gpu/) with an NVIDIA RTX 4000.
|