Sifal commited on
Commit
dc05782
1 Parent(s): 0fbf62a

End of training

Browse files
README.md ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: bert-large-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - glue
8
+ metrics:
9
+ - accuracy
10
+ - f1
11
+ - precision
12
+ - recall
13
+ model-index:
14
+ - name: trainer
15
+ results:
16
+ - task:
17
+ name: Text Classification
18
+ type: text-classification
19
+ dataset:
20
+ name: glue
21
+ type: glue
22
+ config: cola
23
+ split: validation
24
+ args: cola
25
+ metrics:
26
+ - name: Accuracy
27
+ type: accuracy
28
+ value: 0.8465963566634708
29
+ - name: F1
30
+ type: f1
31
+ value: 0.8064540073113251
32
+ - name: Precision
33
+ type: precision
34
+ value: 0.840606542828289
35
+ - name: Recall
36
+ type: recall
37
+ value: 0.7876439727431708
38
+ ---
39
+
40
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
41
+ should probably proofread and complete it, then remove this comment. -->
42
+
43
+ # trainer
44
+
45
+ This model is a fine-tuned version of [bert-large-uncased](https://huggingface.co/bert-large-uncased) on the glue dataset.
46
+ It achieves the following results on the evaluation set:
47
+ - Loss: 0.4490
48
+ - Accuracy: 0.8466
49
+ - F1: 0.8065
50
+ - Precision: 0.8406
51
+ - Recall: 0.7876
52
+
53
+ ## Model description
54
+
55
+ More information needed
56
+
57
+ ## Intended uses & limitations
58
+
59
+ More information needed
60
+
61
+ ## Training and evaluation data
62
+
63
+ More information needed
64
+
65
+ ## Training procedure
66
+
67
+ ### Training hyperparameters
68
+
69
+ The following hyperparameters were used during training:
70
+ - learning_rate: 2e-05
71
+ - train_batch_size: 8
72
+ - eval_batch_size: 32
73
+ - seed: 42
74
+ - gradient_accumulation_steps: 4
75
+ - total_train_batch_size: 32
76
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
77
+ - lr_scheduler_type: linear
78
+ - num_epochs: 2
79
+
80
+ ### Training results
81
+
82
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
83
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
84
+ | No log | 1.0 | 267 | 0.3860 | 0.8370 | 0.7999 | 0.8184 | 0.7876 |
85
+ | 0.3455 | 2.0 | 534 | 0.4490 | 0.8466 | 0.8065 | 0.8406 | 0.7876 |
86
+
87
+
88
+ ### Framework versions
89
+
90
+ - Transformers 4.37.0
91
+ - Pytorch 2.1.2
92
+ - Datasets 2.1.0
93
+ - Tokenizers 0.15.1
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-large-uncased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 1024,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 4096,
14
+ "layer_norm_eps": 1e-12,
15
+ "max_position_embeddings": 512,
16
+ "model_type": "bert",
17
+ "num_attention_heads": 16,
18
+ "num_hidden_layers": 24,
19
+ "pad_token_id": 0,
20
+ "position_embedding_type": "absolute",
21
+ "problem_type": "single_label_classification",
22
+ "torch_dtype": "float32",
23
+ "transformers_version": "4.37.0",
24
+ "type_vocab_size": 2,
25
+ "use_cache": true,
26
+ "vocab_size": 30522
27
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:457c8c6cea33b6283c379c7e0e16ef727edc6165c9cbb4989d5cf4918e39b010
3
+ size 1340622760
runs/Feb10_14-10-25_4c5fee1ac69e/events.out.tfevents.1707574236.4c5fee1ac69e.34.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68068f10afbbdef7ae3480c1edb6d2c41fa105d62dae5931d19b0076b2f87526
3
+ size 5786
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23829543593169706e5540658817bcc14e14793f341e583d257c2810ed0e7afc
3
+ size 4664