Evelyn18 commited on
Commit
76f72f8
1 Parent(s): 162f0f2

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: 5.4867
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: 4e-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.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
 
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.6654
20
 
21
  ## Model description
22
 
 
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
+ - learning_rate: 5e-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.0105 |
51
+ | No log | 2.0 | 10 | 4.2371 |
52
+ | No log | 3.0 | 15 | 4.0368 |
53
+ | No log | 4.0 | 20 | 4.6523 |
54
+ | No log | 5.0 | 25 | 4.9767 |
55
+ | No log | 6.0 | 30 | 5.1290 |
56
+ | No log | 7.0 | 35 | 5.6087 |
57
+ | No log | 8.0 | 40 | 5.5824 |
58
+ | No log | 9.0 | 45 | 5.6266 |
59
+ | No log | 10.0 | 50 | 5.6654 |
60
 
61
 
62
  ### Framework versions