Hugging Face's logo

Chrisantha
/
distilbert-base-uncased-finetuned-imdb

Fill-Mask
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
Chrisantha commited on
Commit
e4b8941
•
1 Parent(s): 3d48651

End of training

Browse files
Files changed (6) hide show
  1. README.md +5 -5
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. runs/May29_10-16-42_7b5aae81e8d3/events.out.tfevents.1716977814.7b5aae81e8d3.1421.0 +3 -0
  5. runs/May29_10-16-42_7b5aae81e8d3/events.out.tfevents.1716977990.7b5aae81e8d3.1421.1 +3 -0
  6. training_args.bin +1 -1
README.md CHANGED
@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 2.4899
19
 
20
  ## Model description
21
 
@@ -47,14 +47,14 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 2.683 | 1.0 | 157 | 2.5116 |
51
- | 2.5932 | 2.0 | 314 | 2.4495 |
52
- | 2.5291 | 3.0 | 471 | 2.4821 |
53
 
54
 
55
  ### Framework versions
56
 
57
- - Transformers 4.41.0
58
  - Pytorch 2.3.0+cu121
59
  - Datasets 2.19.1
60
  - Tokenizers 0.19.1
 
15
 
16
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 2.4894
19
 
20
  ## Model description
21
 
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 2.6819 | 1.0 | 157 | 2.4978 |
51
+ | 2.5872 | 2.0 | 314 | 2.4488 |
52
+ | 2.527 | 3.0 | 471 | 2.4823 |
53
 
54
 
55
  ### Framework versions
56
 
57
+ - Transformers 4.41.1
58
  - Pytorch 2.3.0+cu121
59
  - Datasets 2.19.1
60
  - Tokenizers 0.19.1
config.json CHANGED
@@ -19,6 +19,6 @@
19
  "sinusoidal_pos_embds": false,
20
  "tie_weights_": true,
21
  "torch_dtype": "float32",
22
- "transformers_version": "4.41.0",
23
  "vocab_size": 30522
24
  }
 
19
  "sinusoidal_pos_embds": false,
20
  "tie_weights_": true,
21
  "torch_dtype": "float32",
22
+ "transformers_version": "4.41.1",
23
  "vocab_size": 30522
24
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d017f6fdf7bb1aaa298f9bc36b1c3d235670488817b87c48af7f85dd8c57b78f
3
  size 267954768
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df93f3b2c9b240d47ef5b704507ecd15f2ac6bd6df8ab923b6e68f098c580bc6
3
  size 267954768
runs/May29_10-16-42_7b5aae81e8d3/events.out.tfevents.1716977814.7b5aae81e8d3.1421.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28890210ab382728bbbdda0a3f140f651bded4a5e504f673a4af14593a89acf4
3
+ size 6764
runs/May29_10-16-42_7b5aae81e8d3/events.out.tfevents.1716977990.7b5aae81e8d3.1421.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50d57df0a4873586c2afe40ee07919258cb4bfa2aa08d47ecf855efa5a22fb30
3
+ size 359
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0db378da58a3737f30ffe398e6b5d7b7dc12638bf4a088025730d7e830c7db48
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72f6a9500b3df06e7fc167cfcfe32d8e5f655cc58a35cf5a72029ed3bf6dbd45
3
  size 5176