Evelyn18 commited on
Commit
623029b
1 Parent(s): 3b49c3b

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the becasv2 dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 4.8916
20
 
21
  ## Model description
22
 
@@ -35,7 +35,7 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 1e-05
39
  - train_batch_size: 16
40
  - eval_batch_size: 16
41
  - seed: 42
@@ -47,16 +47,16 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | No log | 1.0 | 5 | 5.8702 |
51
- | No log | 2.0 | 10 | 5.6813 |
52
- | No log | 3.0 | 15 | 5.5046 |
53
- | No log | 4.0 | 20 | 5.3432 |
54
- | No log | 5.0 | 25 | 5.2277 |
55
- | No log | 6.0 | 30 | 5.1122 |
56
- | No log | 7.0 | 35 | 5.0261 |
57
- | No log | 8.0 | 40 | 4.9520 |
58
- | No log | 9.0 | 45 | 4.9076 |
59
- | No log | 10.0 | 50 | 4.8916 |
60
 
61
 
62
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the becasv2 dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 5.4867
20
 
21
  ## Model description
22
 
 
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
+ - learning_rate: 4e-05
39
  - train_batch_size: 16
40
  - eval_batch_size: 16
41
  - seed: 42
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | No log | 1.0 | 5 | 5.0578 |
51
+ | No log | 2.0 | 10 | 4.3346 |
52
+ | No log | 3.0 | 15 | 3.8604 |
53
+ | No log | 4.0 | 20 | 3.8900 |
54
+ | No log | 5.0 | 25 | 4.5578 |
55
+ | No log | 6.0 | 30 | 4.1951 |
56
+ | No log | 7.0 | 35 | 5.3011 |
57
+ | No log | 8.0 | 40 | 5.2158 |
58
+ | No log | 9.0 | 45 | 5.3141 |
59
+ | No log | 10.0 | 50 | 5.4867 |
60
 
61
 
62
  ### Framework versions