icity commited on
Commit
40d17e8
·
1 Parent(s): 4c75359

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -7
README.md CHANGED
@@ -2,8 +2,6 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
- datasets:
6
- - imdb
7
  model-index:
8
  - name: distilbert-base-uncased-finetuned-imdb
9
  results: []
@@ -14,9 +12,9 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # distilbert-base-uncased-finetuned-imdb
16
 
17
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 2.4721
20
 
21
  ## Model description
22
 
@@ -48,9 +46,9 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss |
50
  |:-------------:|:-----:|:----:|:---------------:|
51
- | 2.7086 | 1.0 | 157 | 2.4897 |
52
- | 2.5796 | 2.0 | 314 | 2.4230 |
53
- | 2.5269 | 3.0 | 471 | 2.4354 |
54
 
55
 
56
  ### Framework versions
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  model-index:
6
  - name: distilbert-base-uncased-finetuned-imdb
7
  results: []
 
12
 
13
  # distilbert-base-uncased-finetuned-imdb
14
 
15
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 2.5534
18
 
19
  ## Model description
20
 
 
46
 
47
  | Training Loss | Epoch | Step | Validation Loss |
48
  |:-------------:|:-----:|:----:|:---------------:|
49
+ | 2.9021 | 1.0 | 12 | 2.6918 |
50
+ | 2.7152 | 2.0 | 24 | 2.5909 |
51
+ | 2.6445 | 3.0 | 36 | 2.5104 |
52
 
53
 
54
  ### Framework versions