DerivedFunction commited on
Commit
bdbf613
·
verified ·
1 Parent(s): d5940b6

End of training

Browse files
README.md ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: distilbert/distilbert-base-cased
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: distilbert-imdb
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # distilbert-imdb
18
+
19
+ This model is a fine-tuned version of [distilbert/distilbert-base-cased](https://huggingface.co/distilbert/distilbert-base-cased) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.2111
22
+ - Accuracy: 0.9285
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
+ - train_batch_size: 24
43
+ - eval_batch_size: 24
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 2
48
+ - mixed_precision_training: Native AMP
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
+ |:-------------:|:------:|:----:|:---------------:|:--------:|
54
+ | 0.2040 | 0.4997 | 833 | 0.2493 | 0.8981 |
55
+ | 0.1805 | 0.9994 | 1666 | 0.2134 | 0.915 |
56
+ | 0.1683 | 1.4991 | 2499 | 0.2222 | 0.9262 |
57
+ | 0.1644 | 1.9988 | 3332 | 0.2111 | 0.9285 |
58
+
59
+
60
+ ### Framework versions
61
+
62
+ - Transformers 5.0.0
63
+ - Pytorch 2.10.0+cu128
64
+ - Datasets 4.0.0
65
+ - Tokenizers 0.22.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:284ddaa0fe2e7f2dbd2e8069b46326516082ec8d4ad5705909295d1df9d80402
3
  size 263144680
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54426e5d1f0f5dc02cdf49aa1634e484402d0ecc68a00fc56c0d7df1a19b715b
3
  size 263144680
runs/Feb27_17-16-49_bbfc1e2c4baa/events.out.tfevents.1772212609.bbfc1e2c4baa.1955.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cfac81ab8c30d77d137efcc5e1d79631724852387b7465a8ad265c38afa73b1
3
+ size 76121
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "cls_token": "[CLS]",
4
+ "do_lower_case": false,
5
+ "is_local": false,
6
+ "mask_token": "[MASK]",
7
+ "model_max_length": 512,
8
+ "pad_token": "[PAD]",
9
+ "sep_token": "[SEP]",
10
+ "strip_accents": null,
11
+ "tokenize_chinese_chars": true,
12
+ "tokenizer_class": "BertTokenizer",
13
+ "unk_token": "[UNK]"
14
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8c622feaab55cf7dd5194717ceefc274b0b2b08ca015f4f7eead35cc1d9d6b1c
3
  size 5201
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61adb39d8b1a24a60a1fd297c98c586037f32c31e23c2d86731aa03d3af9f23c
3
  size 5201