NataliaKhaidanova commited on
Commit
bf261bc
·
1 Parent(s): 95175b0

Upload hparams.yaml

Browse files
Files changed (1) hide show
  1. 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