Abubakari commited on
Commit
0207309
1 Parent(s): 67e4e9c

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -8
README.md CHANGED
@@ -14,8 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.6147
18
- - Rmse: 0.6327
19
 
20
  ## Model description
21
 
@@ -34,12 +34,12 @@ More information needed
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
- - learning_rate: 3e-05
38
- - train_batch_size: 4
39
- - eval_batch_size: 4
40
  - seed: 42
41
  - gradient_accumulation_steps: 16
42
- - total_train_batch_size: 64
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_steps: 500
@@ -50,8 +50,11 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
52
  |:-------------:|:-----:|:----:|:---------------:|:------:|
53
- | 0.6405 | 4.0 | 500 | 0.6147 | 0.6327 |
54
- | 0.2214 | 8.0 | 1000 | 0.8795 | 0.6251 |
 
 
 
55
 
56
 
57
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 0.5618
18
+ - Rmse: 0.6118
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - learning_rate: 2e-05
38
+ - train_batch_size: 2
39
+ - eval_batch_size: 2
40
  - seed: 42
41
  - gradient_accumulation_steps: 16
42
+ - total_train_batch_size: 32
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_steps: 500
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
52
  |:-------------:|:-----:|:----:|:---------------:|:------:|
53
+ | 0.7273 | 2.0 | 500 | 0.5618 | 0.6118 |
54
+ | 0.4294 | 4.0 | 1000 | 0.5821 | 0.5906 |
55
+ | 0.2278 | 6.0 | 1500 | 0.8019 | 0.6235 |
56
+ | 0.1246 | 8.0 | 2000 | 0.9412 | 0.5961 |
57
+ | 0.083 | 10.0 | 2500 | 1.1040 | 0.5978 |
58
 
59
 
60
  ### Framework versions