era28 commited on
Commit
77b76ca
·
1 Parent(s): e078443

Upload Referenceless Regression model checkpoint

Browse files
Files changed (3) hide show
  1. checkpoints/model.ckpt +3 -0
  2. hparams.yaml +37 -0
  3. metrics.csv +0 -0
checkpoints/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6461e5ee86e4200fd85ee30b6078f4611eb2ba3364a3ba6ffbcff6c8158b2f3d
3
+ size 2277503524
hparams.yaml ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ activations: Tanh
2
+ batch_size: 16
3
+ class_identifier: referenceless_regression_metric
4
+ dropout: 0.1
5
+ encoder_learning_rate: 1.0e-06
6
+ encoder_model: XLM-RoBERTa
7
+ final_activation: null
8
+ hidden_sizes:
9
+ - 2048
10
+ - 1024
11
+ keep_embeddings_frozen: true
12
+ layer: mix
13
+ layer_norm: false
14
+ layer_transformation: sparsemax
15
+ layerwise_decay: 0.95
16
+ learning_rate: 1.5e-05
17
+ load_pretrained_weights: true
18
+ local_files_only: false
19
+ loss: mse
20
+ nr_frozen_epochs: 0.3
21
+ optimizer: AdamW
22
+ pool: avg
23
+ pretrained_model: xlm-roberta-large
24
+ train_data:
25
+ - data/combined_dataset.csv
26
+ validation_data:
27
+ - data/xcomet_noref_Chinese_test.csv
28
+ - data/xcomet_noref_Czech_test.csv
29
+ - data/xcomet_noref_Estonian_test.csv
30
+ - data/xcomet_noref_Finnish_test.csv
31
+ - data/xcomet_noref_German_test.csv
32
+ - data/xcomet_noref_Hausa_test.csv
33
+ - data/xcomet_noref_Japanese_test.csv
34
+ - data/xcomet_noref_Lithuanian_test.csv
35
+ - data/xcomet_noref_Russian_test.csv
36
+ - data/xcomet_noref_Turkish_test.csv
37
+ warmup_steps: 0
metrics.csv ADDED
The diff for this file is too large to render. See raw diff