anton-l HF staff commited on
Commit
c2b08c8
1 Parent(s): 13e3e6f

Model save

Browse files
README.md ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: intfloat/multilingual-e5-small
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - precision
9
+ - recall
10
+ - accuracy
11
+ model-index:
12
+ - name: owm-math-scorer-multilingual-e5-small
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # owm-math-scorer-multilingual-e5-small
20
+
21
+ This model is a fine-tuned version of [intfloat/multilingual-e5-small](https://huggingface.co/intfloat/multilingual-e5-small) on an unknown dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.4478
24
+ - Precision: 0.8771
25
+ - Recall: 0.8769
26
+ - F1 Macro: 0.8770
27
+ - Accuracy: 0.8770
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 0.001
47
+ - train_batch_size: 256
48
+ - eval_batch_size: 128
49
+ - seed: 0
50
+ - distributed_type: multi-GPU
51
+ - num_devices: 4
52
+ - total_train_batch_size: 1024
53
+ - total_eval_batch_size: 512
54
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
+ - lr_scheduler_type: linear
56
+ - num_epochs: 20
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 Macro | Accuracy |
61
+ |:-------------:|:-------:|:-----:|:---------------:|:---------:|:------:|:--------:|:--------:|
62
+ | No log | 0 | 0 | 8.2909 | 0.2546 | 0.5 | 0.3374 | 0.5091 |
63
+ | 0.5633 | 0.2844 | 250 | 0.5661 | 0.8608 | 0.8596 | 0.8588 | 0.8589 |
64
+ | 0.5443 | 0.5688 | 500 | 0.5192 | 0.8655 | 0.8652 | 0.8653 | 0.8654 |
65
+ | 0.5395 | 0.8532 | 750 | 0.5461 | 0.8651 | 0.8636 | 0.8628 | 0.8629 |
66
+ | 0.5144 | 1.1377 | 1000 | 0.4992 | 0.8691 | 0.8692 | 0.8691 | 0.8691 |
67
+ | 0.5278 | 1.4221 | 1250 | 0.5322 | 0.8675 | 0.8613 | 0.8616 | 0.8624 |
68
+ | 0.501 | 1.7065 | 1500 | 0.4942 | 0.8708 | 0.8690 | 0.8692 | 0.8695 |
69
+ | 0.4942 | 1.9909 | 1750 | 0.4934 | 0.8697 | 0.8696 | 0.8693 | 0.8693 |
70
+ | 0.492 | 2.2753 | 2000 | 0.4873 | 0.8710 | 0.8711 | 0.8711 | 0.8711 |
71
+ | 0.4984 | 2.5597 | 2250 | 0.5061 | 0.8701 | 0.8694 | 0.8688 | 0.8688 |
72
+ | 0.4809 | 2.8441 | 2500 | 0.4995 | 0.8719 | 0.8673 | 0.8677 | 0.8682 |
73
+ | 0.4744 | 3.1286 | 2750 | 0.4783 | 0.8721 | 0.8722 | 0.8721 | 0.8721 |
74
+ | 0.4817 | 3.4130 | 3000 | 0.4715 | 0.8737 | 0.8738 | 0.8738 | 0.8738 |
75
+ | 0.4748 | 3.6974 | 3250 | 0.4734 | 0.8743 | 0.8725 | 0.8728 | 0.8731 |
76
+ | 0.4725 | 3.9818 | 3500 | 0.4703 | 0.8738 | 0.8736 | 0.8737 | 0.8738 |
77
+ | 0.4684 | 4.2662 | 3750 | 0.4693 | 0.8739 | 0.8734 | 0.8735 | 0.8737 |
78
+ | 0.4796 | 4.5506 | 4000 | 0.4697 | 0.8746 | 0.8727 | 0.8729 | 0.8732 |
79
+ | 0.4666 | 4.8350 | 4250 | 0.4715 | 0.8737 | 0.8738 | 0.8735 | 0.8735 |
80
+ | 0.4697 | 5.1195 | 4500 | 0.4853 | 0.8736 | 0.8692 | 0.8695 | 0.8700 |
81
+ | 0.466 | 5.4039 | 4750 | 0.4782 | 0.8734 | 0.8713 | 0.8716 | 0.8719 |
82
+ | 0.4663 | 5.6883 | 5000 | 0.4653 | 0.8746 | 0.8747 | 0.8746 | 0.8746 |
83
+ | 0.4677 | 5.9727 | 5250 | 0.4656 | 0.8749 | 0.8734 | 0.8737 | 0.8739 |
84
+ | 0.4615 | 6.2571 | 5500 | 0.4631 | 0.8753 | 0.8739 | 0.8741 | 0.8743 |
85
+ | 0.4689 | 6.5415 | 5750 | 0.4610 | 0.8759 | 0.8754 | 0.8756 | 0.8757 |
86
+ | 0.4643 | 6.8259 | 6000 | 0.4601 | 0.8753 | 0.8747 | 0.8749 | 0.8750 |
87
+ | 0.4591 | 7.1104 | 6250 | 0.4598 | 0.8748 | 0.8745 | 0.8746 | 0.8747 |
88
+ | 0.4628 | 7.3948 | 6500 | 0.4592 | 0.8759 | 0.8749 | 0.8751 | 0.8753 |
89
+ | 0.4589 | 7.6792 | 6750 | 0.4613 | 0.8759 | 0.8744 | 0.8747 | 0.8749 |
90
+ | 0.4626 | 7.9636 | 7000 | 0.4566 | 0.8758 | 0.8753 | 0.8754 | 0.8756 |
91
+ | 0.4632 | 8.2480 | 7250 | 0.4623 | 0.8746 | 0.8727 | 0.8730 | 0.8732 |
92
+ | 0.4545 | 8.5324 | 7500 | 0.4554 | 0.8766 | 0.8759 | 0.8760 | 0.8762 |
93
+ | 0.4596 | 8.8168 | 7750 | 0.4581 | 0.8755 | 0.8755 | 0.8755 | 0.8755 |
94
+ | 0.4571 | 9.1013 | 8000 | 0.4595 | 0.8759 | 0.8737 | 0.8740 | 0.8743 |
95
+ | 0.4585 | 9.3857 | 8250 | 0.4561 | 0.8760 | 0.8750 | 0.8752 | 0.8754 |
96
+ | 0.4541 | 9.6701 | 8500 | 0.4548 | 0.8756 | 0.8750 | 0.8751 | 0.8752 |
97
+ | 0.4576 | 9.9545 | 8750 | 0.4541 | 0.8757 | 0.8754 | 0.8755 | 0.8756 |
98
+ | 0.449 | 10.2389 | 9000 | 0.4554 | 0.8754 | 0.8752 | 0.8752 | 0.8753 |
99
+ | 0.4507 | 10.5233 | 9250 | 0.4535 | 0.8763 | 0.8763 | 0.8763 | 0.8763 |
100
+ | 0.4545 | 10.8077 | 9500 | 0.4543 | 0.8759 | 0.8758 | 0.8758 | 0.8759 |
101
+ | 0.4462 | 11.0922 | 9750 | 0.4529 | 0.8764 | 0.8756 | 0.8758 | 0.8759 |
102
+ | 0.4505 | 11.3766 | 10000 | 0.4538 | 0.8762 | 0.8751 | 0.8753 | 0.8755 |
103
+ | 0.4576 | 11.6610 | 10250 | 0.4714 | 0.8751 | 0.8714 | 0.8717 | 0.8722 |
104
+ | 0.4509 | 11.9454 | 10500 | 0.4613 | 0.8759 | 0.8760 | 0.8758 | 0.8758 |
105
+ | 0.4557 | 12.2298 | 10750 | 0.4538 | 0.8764 | 0.8753 | 0.8755 | 0.8757 |
106
+ | 0.4539 | 12.5142 | 11000 | 0.4523 | 0.8765 | 0.8758 | 0.8760 | 0.8761 |
107
+ | 0.4534 | 12.7986 | 11250 | 0.4515 | 0.8766 | 0.8767 | 0.8766 | 0.8767 |
108
+ | 0.4532 | 13.0830 | 11500 | 0.4509 | 0.8768 | 0.8763 | 0.8765 | 0.8766 |
109
+ | 0.4501 | 13.3675 | 11750 | 0.4517 | 0.8765 | 0.8762 | 0.8763 | 0.8763 |
110
+ | 0.4493 | 13.6519 | 12000 | 0.4527 | 0.8767 | 0.8768 | 0.8768 | 0.8768 |
111
+ | 0.4528 | 13.9363 | 12250 | 0.4499 | 0.8766 | 0.8765 | 0.8765 | 0.8766 |
112
+ | 0.4491 | 14.2207 | 12500 | 0.4519 | 0.8766 | 0.8755 | 0.8757 | 0.8759 |
113
+ | 0.4495 | 14.5051 | 12750 | 0.4594 | 0.8768 | 0.8769 | 0.8767 | 0.8767 |
114
+ | 0.4443 | 14.7895 | 13000 | 0.4519 | 0.8766 | 0.8764 | 0.8765 | 0.8766 |
115
+ | 0.4476 | 15.0739 | 13250 | 0.4509 | 0.8769 | 0.8766 | 0.8767 | 0.8768 |
116
+ | 0.4466 | 15.3584 | 13500 | 0.4494 | 0.8773 | 0.8769 | 0.8770 | 0.8771 |
117
+ | 0.4456 | 15.6428 | 13750 | 0.4489 | 0.8768 | 0.8765 | 0.8766 | 0.8767 |
118
+ | 0.4447 | 15.9272 | 14000 | 0.4552 | 0.8765 | 0.8751 | 0.8754 | 0.8756 |
119
+ | 0.4471 | 16.2116 | 14250 | 0.4520 | 0.8763 | 0.8763 | 0.8763 | 0.8763 |
120
+ | 0.4521 | 16.4960 | 14500 | 0.4509 | 0.8770 | 0.8756 | 0.8758 | 0.8760 |
121
+ | 0.4419 | 16.7804 | 14750 | 0.4533 | 0.8767 | 0.8768 | 0.8767 | 0.8768 |
122
+ | 0.4485 | 17.0648 | 15000 | 0.4483 | 0.8770 | 0.8768 | 0.8769 | 0.8769 |
123
+ | 0.4424 | 17.3493 | 15250 | 0.4490 | 0.8770 | 0.8769 | 0.8769 | 0.8770 |
124
+ | 0.4441 | 17.6337 | 15500 | 0.4502 | 0.8770 | 0.8769 | 0.8770 | 0.8770 |
125
+ | 0.4487 | 17.9181 | 15750 | 0.4480 | 0.8769 | 0.8763 | 0.8765 | 0.8766 |
126
+ | 0.4487 | 18.2025 | 16000 | 0.4500 | 0.8771 | 0.8772 | 0.8772 | 0.8772 |
127
+ | 0.4375 | 18.4869 | 16250 | 0.4483 | 0.8769 | 0.8766 | 0.8767 | 0.8768 |
128
+ | 0.4491 | 18.7713 | 16500 | 0.4515 | 0.8768 | 0.8769 | 0.8768 | 0.8768 |
129
+ | 0.4433 | 19.0557 | 16750 | 0.4477 | 0.8773 | 0.8769 | 0.8770 | 0.8771 |
130
+ | 0.4432 | 19.3402 | 17000 | 0.4480 | 0.8771 | 0.8769 | 0.8770 | 0.8771 |
131
+ | 0.442 | 19.6246 | 17250 | 0.4480 | 0.8770 | 0.8768 | 0.8769 | 0.8770 |
132
+ | 0.4407 | 19.9090 | 17500 | 0.4478 | 0.8771 | 0.8769 | 0.8770 | 0.8770 |
133
+
134
+
135
+ ### Framework versions
136
+
137
+ - Transformers 4.44.2
138
+ - Pytorch 2.4.0+cu121
139
+ - Datasets 2.21.0
140
+ - Tokenizers 0.19.1
final/config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "intfloat/multilingual-e5-small",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": 0.0,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 384,
11
+ "id2label": {
12
+ "0": "LABEL_0"
13
+ },
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 1536,
16
+ "label2id": {
17
+ "LABEL_0": 0
18
+ },
19
+ "layer_norm_eps": 1e-12,
20
+ "max_position_embeddings": 512,
21
+ "model_type": "bert",
22
+ "num_attention_heads": 12,
23
+ "num_hidden_layers": 12,
24
+ "pad_token_id": 0,
25
+ "position_embedding_type": "absolute",
26
+ "problem_type": "regression",
27
+ "tokenizer_class": "XLMRobertaTokenizer",
28
+ "torch_dtype": "float32",
29
+ "transformers_version": "4.44.2",
30
+ "type_vocab_size": 2,
31
+ "use_cache": true,
32
+ "vocab_size": 250037
33
+ }
final/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bad3fbea5207a34479f4068e5f44866114033fe6ee4fc86ff4d850885c49f842
3
+ size 470640124
final/sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
3
+ size 5069051
final/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
final/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd98e5698b201ba914efb8c18b6709fa8735ab71dcad8d2b431e52e8bf68d932
3
+ size 17082800
final/tokenizer_config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "250001": {
36
+ "content": "<mask>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "<s>",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "<s>",
47
+ "eos_token": "</s>",
48
+ "mask_token": "<mask>",
49
+ "model_max_length": 512,
50
+ "pad_token": "<pad>",
51
+ "sep_token": "</s>",
52
+ "sp_model_kwargs": {},
53
+ "tokenizer_class": "XLMRobertaTokenizer",
54
+ "unk_token": "<unk>"
55
+ }
final/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ce87380f1da8b14836c6c05d1736e7aed753fed6d93fdf122bcf0d6c2242717
3
+ size 5304
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1aa1d5f613327c4870b06b3bfa34441cb0357510834f74ac4817f73372a6a79d
3
  size 470640124
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bad3fbea5207a34479f4068e5f44866114033fe6ee4fc86ff4d850885c49f842
3
  size 470640124