Initial commit
Browse files- .gitattributes +2 -0
- checkpoints/model.ckpt +3 -0
- hparams.yaml +21 -0
.gitattributes
CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
checkpoints/model.ckpt filter=lfs diff=lfs merge=lfs -text
|
37 |
+
|
checkpoints/model.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eac5ce4cd3a34bf6d90a2b3ab4e73c2589b92bb32bc17b00c276629b2ecf06e4
|
3 |
+
size 2277430785
|
hparams.yaml
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# seed 8
|
2 |
+
activations: Tanh
|
3 |
+
batch_size: 4
|
4 |
+
class_identifier: referenceless_regression_metric
|
5 |
+
dropout: 0.15
|
6 |
+
encoder_learning_rate: 1.0e-05
|
7 |
+
encoder_model: XLM-RoBERTa
|
8 |
+
hidden_sizes:
|
9 |
+
- 2048
|
10 |
+
- 1024
|
11 |
+
keep_embeddings_frozen: true
|
12 |
+
layer: mix
|
13 |
+
layerwise_decay: 0.95
|
14 |
+
learning_rate: 1.0e-05
|
15 |
+
load_weights_from_checkpoint: null
|
16 |
+
nr_frozen_epochs: 1
|
17 |
+
optimizer: AdamW
|
18 |
+
pool: avg
|
19 |
+
pretrained_model: xlm-roberta-large
|
20 |
+
train_data: data/mqm.train.z_score.csv
|
21 |
+
validation_data: data/mqm.test.z_score.csv
|