readme: add dataset information to metadata
Browse filesHey,
this PR adds the used dataset to metadata tag.
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.
|