wyzhw commited on
Commit
6911c16
1 Parent(s): 1d8082b

End of training

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -3,8 +3,6 @@ license: apache-2.0
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
6
- datasets:
7
- - imdb
8
  model-index:
9
  - name: N_distilbert_imdb_padding10model
10
  results: []
@@ -15,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # N_distilbert_imdb_padding10model
17
 
18
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
19
 
20
  ## Model description
21
 
@@ -46,12 +44,12 @@ The following hyperparameters were used during training:
46
 
47
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
48
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
49
- | No log | 0.01 | 16 | 0.6795 | 0.6919 |
50
 
51
 
52
  ### Framework versions
53
 
54
- - Transformers 4.35.0
55
- - Pytorch 2.0.0+cu117
56
- - Datasets 2.14.6
57
- - Tokenizers 0.14.1
 
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
 
 
6
  model-index:
7
  - name: N_distilbert_imdb_padding10model
8
  results: []
 
13
 
14
  # N_distilbert_imdb_padding10model
15
 
16
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
17
 
18
  ## Model description
19
 
 
44
 
45
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
46
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
47
+ | No log | 0.01 | 16 | 0.6821 | 0.7085 |
48
 
49
 
50
  ### Framework versions
51
 
52
+ - Transformers 4.37.2
53
+ - Pytorch 2.2.0
54
+ - Datasets 2.17.0
55
+ - Tokenizers 0.15.2