DerivedFunction commited on
Commit
c69f08c
·
verified ·
1 Parent(s): 11f87d9

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: albert/albert-base-v2
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: albert-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
+ # albert-imdb
18
+
19
+ This model is a fine-tuned version of [albert/albert-base-v2](https://huggingface.co/albert/albert-base-v2) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.1795
22
+ - Accuracy: 0.9479
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.2346 | 0.4997 | 833 | 0.1875 | 0.9291 |
55
+ | 0.1219 | 0.9994 | 1666 | 0.2007 | 0.9305 |
56
+ | 0.1825 | 1.4991 | 2499 | 0.2000 | 0.9464 |
57
+ | 0.1559 | 1.9988 | 3332 | 0.1795 | 0.9479 |
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:f8055c3a4a96857b110b9cd511b5ecd7fd901870d65a215dc507243f158f2196
3
  size 46743904
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cfce5a9e13027c90024096abbf60cf161ec15aa362dbad4833663d8b657be50
3
  size 46743904
runs/Feb27_16-12-25_bbfc1e2c4baa/events.out.tfevents.1772208745.bbfc1e2c4baa.1955.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e037f4c200068e8344d3c25d44ca236e7360a242df6df020742b7327d06a6e6c
3
+ size 76288
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "bos_token": "[CLS]",
5
+ "cls_token": "[CLS]",
6
+ "do_lower_case": true,
7
+ "eos_token": "[SEP]",
8
+ "is_local": false,
9
+ "keep_accents": false,
10
+ "mask_token": "[MASK]",
11
+ "model_max_length": 512,
12
+ "pad_token": "<pad>",
13
+ "sep_token": "[SEP]",
14
+ "tokenizer_class": "AlbertTokenizer",
15
+ "trim_offsets": true,
16
+ "unk_token": "<unk>"
17
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:904088724298216ddd600e639d6d9b3e542d1057b4a5ee4b8ad3570729322f88
3
  size 5201
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e952f1433291e81f3569145140f97bd722c23eb96179c1d2abf71c1d42863231
3
  size 5201