Commit
·
bf261bc
1
Parent(s):
95175b0
Upload hparams.yaml
Browse files- hparams.yaml +20 -0
hparams.yaml
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
nr_frozen_epochs: 1
|
2 |
+
keep_embeddings_frozen: true
|
3 |
+
optimizer: AdamW
|
4 |
+
encoder_learning_rate: 1.0e-05
|
5 |
+
learning_rate: 1.0e-05
|
6 |
+
layerwise_decay: 0.95
|
7 |
+
encoder_model: XLM-RoBERTa
|
8 |
+
pretrained_model: xlm-roberta-large
|
9 |
+
pool: avg
|
10 |
+
layer: mix
|
11 |
+
dropout: 0.15
|
12 |
+
batch_size: 4
|
13 |
+
class_identifier: regression_metric
|
14 |
+
train_data: data/mqm.train.z_score.csv
|
15 |
+
validation_data: data/mqm.test.z_score.csv
|
16 |
+
hidden_sizes:
|
17 |
+
- 3072
|
18 |
+
- 1024
|
19 |
+
activations: Tanh
|
20 |
+
load_weights_from_checkpoint: null
|