Arihant Tripathi commited on
Commit
5fd3441
·
verified ·
1 Parent(s): 5f00aba

qwen_new_mage_per_domain_balanced_moe_c10

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: other
4
+ base_model: Qwen/Qwen1.5-MoE-A2.7B
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: fine_tuned_per_domain_balanced_moe_c10
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
+ # fine_tuned_per_domain_balanced_moe_c10
18
+
19
+ This model is a fine-tuned version of [Qwen/Qwen1.5-MoE-A2.7B](https://huggingface.co/Qwen/Qwen1.5-MoE-A2.7B) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 2.2149
22
+ - Accuracy: 0.5374
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: 1
43
+ - eval_batch_size: 1
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH 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: 3
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
52
+ |:-------------:|:------:|:----:|:--------:|:---------------:|
53
+ | 7.9537 | 0.0006 | 100 | 0.5384 | 4.2406 |
54
+ | 2.7142 | 0.0013 | 200 | 0.5386 | 6.1312 |
55
+ | 2.5969 | 0.0019 | 300 | 0.4651 | 1.0811 |
56
+ | 3.6087 | 0.0025 | 400 | 0.4655 | 1.7135 |
57
+ | 3.217 | 0.0032 | 500 | 0.5386 | 2.4567 |
58
+ | 2.0844 | 0.0038 | 600 | 0.4614 | 3.8137 |
59
+ | 3.0955 | 0.0044 | 700 | 0.5386 | 1.2668 |
60
+ | 2.0157 | 0.0051 | 800 | 0.5386 | 3.2796 |
61
+ | 2.4513 | 0.0057 | 900 | 0.4614 | 2.2765 |
62
+ | 2.482 | 0.0063 | 1000 | 0.5386 | 0.7492 |
63
+ | 2.3079 | 0.0070 | 1100 | 0.5386 | 1.6933 |
64
+ | 2.5698 | 0.0076 | 1200 | 0.5386 | 3.1721 |
65
+ | 2.4214 | 0.0082 | 1300 | 0.5386 | 1.7702 |
66
+ | 1.2708 | 0.0089 | 1400 | 0.4646 | 0.9111 |
67
+ | 0.8665 | 0.0095 | 1500 | 0.5494 | 0.6819 |
68
+ | 1.7844 | 0.0101 | 1600 | 0.5386 | 1.7757 |
69
+ | 2.9675 | 0.0108 | 1700 | 0.5386 | 2.7387 |
70
+ | 2.7119 | 0.0114 | 1800 | 0.5386 | 2.6287 |
71
+ | 2.526 | 0.0120 | 1900 | 0.5386 | 1.4967 |
72
+ | 3.2745 | 0.0127 | 2000 | 0.4614 | 4.2874 |
73
+ | 3.4052 | 0.0133 | 2100 | 1.0082 | 0.4624 |
74
+ | 1.7179 | 0.0139 | 2200 | 1.6046 | 0.4666 |
75
+ | 2.7225 | 0.0146 | 2300 | 3.3510 | 0.5376 |
76
+ | 2.2919 | 0.0152 | 2400 | 3.3149 | 0.5376 |
77
+ | 1.729 | 0.0158 | 2500 | 2.1687 | 0.5376 |
78
+ | 2.5072 | 0.0165 | 2600 | 2.9068 | 0.5376 |
79
+ | 1.9138 | 0.0171 | 2700 | 1.4200 | 0.4624 |
80
+ | 1.4881 | 0.0177 | 2800 | 2.2129 | 0.4631 |
81
+ | 2.031 | 0.0184 | 2900 | 2.2580 | 0.5370 |
82
+ | 1.998 | 0.0190 | 3000 | 2.2149 | 0.5374 |
83
+
84
+
85
+ ### Framework versions
86
+
87
+ - Transformers 4.49.0
88
+ - Pytorch 2.6.0+cu126
89
+ - Datasets 3.3.2
90
+ - Tokenizers 0.21.0
added_tokens.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "<|endoftext|>": 151643,
3
+ "<|im_end|>": 151645,
4
+ "<|im_start|>": 151644
5
+ }
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Qwen/Qwen1.5-MoE-A2.7B",
3
+ "architectures": [
4
+ "Qwen2MoeForSequenceClassification"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "decoder_sparse_step": 1,
9
+ "eos_token_id": 151643,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 2048,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 5632,
14
+ "max_position_embeddings": 8192,
15
+ "max_window_layers": 21,
16
+ "mlp_only_layers": [],
17
+ "model_type": "qwen2_moe",
18
+ "moe_intermediate_size": 1408,
19
+ "norm_topk_prob": false,
20
+ "num_attention_heads": 16,
21
+ "num_experts": 60,
22
+ "num_experts_per_tok": 4,
23
+ "num_hidden_layers": 24,
24
+ "num_key_value_heads": 16,
25
+ "output_router_logits": false,
26
+ "pad_token_id": 151643,
27
+ "problem_type": "single_label_classification",
28
+ "rms_norm_eps": 1e-06,
29
+ "rope_scaling": null,
30
+ "rope_theta": 1000000.0,
31
+ "router_aux_loss_coef": 0.001,
32
+ "shared_expert_intermediate_size": 5632,
33
+ "sliding_window": null,
34
+ "tie_word_embeddings": false,
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.49.0",
37
+ "use_cache": true,
38
+ "use_sliding_window": false,
39
+ "vocab_size": 151936
40
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model-00001-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0949c0e69342fc6ec5dc56e2007d5dae8b1340c9f430042ad055b02b4b6c637
3
+ size 4990221104
model-00002-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e84fd413ac1d660779e391e299133362f37fbad52d502809382f2e6f0db609a6
3
+ size 4991306528
model-00003-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5104bf21ea5b55553d274a2ad602021443e414bfb38393cf3207fd285743608e
3
+ size 4990298240
model-00004-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f78d83cca86a2b8e3743d5d4d327c23afd8eb1dcd72fa3a0cf7c6496a4ba6a4e
3
+ size 4990757696
model-00005-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd05f558cc20d02a02d957aab86727bb3e34b051cb7e88a194876043630e955b
3
+ size 4991306600
model-00006-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ea88a0630f5feb5b95c1f8d3ab48fcfdf275716224cba136f03ea6d2d3670ec
3
+ size 4991306936
model-00007-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1f02f8c5a1f06714f2be735f3ebd19988ad5bf43530db35731933189183298b
3
+ size 4991306952
model-00008-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b1b41c5f0ef04b056b85f78d0b4e7c39125b27c89bf94483c262062d4e979d2
3
+ size 4968238032
model-00009-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82efdd1128051870bed853bc2c9987f329ac3cfbe3b3cf23d3879bc5c9ca7bb9
3
+ size 4989749864
model-00010-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:152f919550b80e212f29d1d86e75e9dae8ddcbdd4238c9d056593df6b0227379
3
+ size 4991306928
model-00011-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5333aa18a28c7a1168df97e5553e8bb2b849ca39d645a2a596be64da28a96355
3
+ size 4991306936
model-00012-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:450e4efb1138eb53ad0bf9765658bc2661718a385e7a776ee65a023f3ff08c23
3
+ size 1141959976
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
special_tokens_map.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>"
5
+ ],
6
+ "eos_token": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false
12
+ },
13
+ "pad_token": {
14
+ "content": "<|endoftext|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false
19
+ }
20
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5ae291beb14c5986404b36fdaabbcf3cb858121c6239ffbec6c1405cbafd9dc
3
+ size 11418544
tokenizer_config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "151643": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "151644": {
13
+ "content": "<|im_start|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "151645": {
21
+ "content": "<|im_end|>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ }
28
+ },
29
+ "additional_special_tokens": [
30
+ "<|im_start|>",
31
+ "<|im_end|>"
32
+ ],
33
+ "bos_token": null,
34
+ "chat_template": "{% for message in messages %}{% if loop.first and messages[0]['role'] != 'system' %}{{ '<|im_start|>system\nYou are a helpful assistant<|im_end|>\n' }}{% endif %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
35
+ "clean_up_tokenization_spaces": false,
36
+ "eos_token": "<|endoftext|>",
37
+ "errors": "replace",
38
+ "extra_special_tokens": {},
39
+ "model_max_length": 32768,
40
+ "pad_token": "<|endoftext|>",
41
+ "split_special_tokens": false,
42
+ "tokenizer_class": "Qwen2Tokenizer",
43
+ "unk_token": null
44
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35c6d1b4bcc10c81455becaa2571094d6ea215e4411e0d29db27556b7dd26dd9
3
+ size 5304
vocab.json ADDED
The diff for this file is too large to render. See raw diff