DerivedFunction commited on
Commit
d09cf15
·
verified ·
1 Parent(s): e7af3b6

End of training

Browse files
README.md ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: FacebookAI/roberta-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: roberta-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
+ # roberta-imdb
18
+
19
+ This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.1680
22
+ - Accuracy: 0.9561
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.2052 | 0.4997 | 833 | 0.1646 | 0.9435 |
55
+ | 0.1565 | 0.9994 | 1666 | 0.1862 | 0.9421 |
56
+ | 0.2213 | 1.4991 | 2499 | 0.1759 | 0.9547 |
57
+ | 0.1314 | 1.9988 | 3332 | 0.1680 | 0.9561 |
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:e98e728f2b447147c5ffe58dbefcd92d7b8a60b0a49bfc9d579c732e308edd02
3
  size 498612800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce15292532e7dc8befdad63477d0d6c0af989685548b23f007ad69560a0c7b7c
3
  size 498612800
runs/Feb27_18-53-14_bbfc1e2c4baa/events.out.tfevents.1772218394.bbfc1e2c4baa.1955.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:934185f13daef3587651e229d8dfb40574658c506caceaca7ec49c82291b5b0d
3
+ size 76171
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<s>",
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "errors": "replace",
8
+ "is_local": false,
9
+ "mask_token": "<mask>",
10
+ "model_max_length": 512,
11
+ "pad_token": "<pad>",
12
+ "sep_token": "</s>",
13
+ "tokenizer_class": "RobertaTokenizer",
14
+ "trim_offsets": true,
15
+ "unk_token": "<unk>"
16
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a480092bb8809d9af2253c401fcd5792dfe2f6119f8bb88895ac648c8e5021fd
3
  size 5201
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a9dc302a36a7af4feda95ea82d8fbf031597604cb2c915690220bc3098f0780
3
  size 5201