PakNin commited on
Commit
d9dbdca
·
verified ·
1 Parent(s): 8755629

Upload folder using huggingface_hub

Browse files
epoch_1/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.18.1
epoch_1/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "q_proj",
33
+ "w2",
34
+ "v_proj",
35
+ "k_proj",
36
+ "o_proj",
37
+ "w3",
38
+ "w1"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
epoch_1/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf182932a5d8839857fa2f19df686dfcd73b516e70cfaef4a8ae6df4afae5059
3
+ size 179902200
epoch_1/chat_template.jinja ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {% for message in messages %}{{'<|' + message['role'] + '|>' + '
2
+ ' + message['content'] + '<|end|>
3
+ ' }}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>
4
+ ' }}{% else %}{{ eos_token }}{% endif %}
epoch_1/rexmoe_routers.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97ed82476c09051f842db77d3c1675ea63a48588510368c86402da2b8e07f642
3
+ size 12619762
epoch_1/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
epoch_1/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
epoch_1/tokenizer_config.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": true,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "32000": {
31
+ "content": "<|endoftext|>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": true
37
+ },
38
+ "32001": {
39
+ "content": "<|assistant|>",
40
+ "lstrip": false,
41
+ "normalized": false,
42
+ "rstrip": true,
43
+ "single_word": false,
44
+ "special": true
45
+ },
46
+ "32002": {
47
+ "content": "<|placeholder1|>",
48
+ "lstrip": false,
49
+ "normalized": false,
50
+ "rstrip": true,
51
+ "single_word": false,
52
+ "special": true
53
+ },
54
+ "32003": {
55
+ "content": "<|placeholder2|>",
56
+ "lstrip": false,
57
+ "normalized": false,
58
+ "rstrip": true,
59
+ "single_word": false,
60
+ "special": true
61
+ },
62
+ "32004": {
63
+ "content": "<|placeholder3|>",
64
+ "lstrip": false,
65
+ "normalized": false,
66
+ "rstrip": true,
67
+ "single_word": false,
68
+ "special": true
69
+ },
70
+ "32005": {
71
+ "content": "<|placeholder4|>",
72
+ "lstrip": false,
73
+ "normalized": false,
74
+ "rstrip": true,
75
+ "single_word": false,
76
+ "special": true
77
+ },
78
+ "32006": {
79
+ "content": "<|system|>",
80
+ "lstrip": false,
81
+ "normalized": false,
82
+ "rstrip": true,
83
+ "single_word": false,
84
+ "special": true
85
+ },
86
+ "32007": {
87
+ "content": "<|end|>",
88
+ "lstrip": false,
89
+ "normalized": false,
90
+ "rstrip": true,
91
+ "single_word": false,
92
+ "special": true
93
+ },
94
+ "32008": {
95
+ "content": "<|placeholder5|>",
96
+ "lstrip": false,
97
+ "normalized": false,
98
+ "rstrip": true,
99
+ "single_word": false,
100
+ "special": true
101
+ },
102
+ "32009": {
103
+ "content": "<|placeholder6|>",
104
+ "lstrip": false,
105
+ "normalized": false,
106
+ "rstrip": true,
107
+ "single_word": false,
108
+ "special": true
109
+ },
110
+ "32010": {
111
+ "content": "<|user|>",
112
+ "lstrip": false,
113
+ "normalized": false,
114
+ "rstrip": true,
115
+ "single_word": false,
116
+ "special": true
117
+ }
118
+ },
119
+ "bos_token": "<s>",
120
+ "clean_up_tokenization_spaces": false,
121
+ "eos_token": "<|endoftext|>",
122
+ "extra_special_tokens": {},
123
+ "legacy": false,
124
+ "model_max_length": 4096,
125
+ "pad_token": "<|endoftext|>",
126
+ "padding_side": "left",
127
+ "sp_model_kwargs": {},
128
+ "tokenizer_class": "LlamaTokenizerFast",
129
+ "unk_token": "<unk>",
130
+ "use_default_system_prompt": false
131
+ }
epoch_2/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.18.1
epoch_2/adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "q_proj",
33
+ "w2",
34
+ "v_proj",
35
+ "k_proj",
36
+ "o_proj",
37
+ "w3",
38
+ "w1"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
epoch_2/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:814ee1e3fe24e3bfb43c350bfbf6eca60f6457148e33ef880a1e5326b5e06e14
3
+ size 179902200
epoch_2/chat_template.jinja ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {% for message in messages %}{{'<|' + message['role'] + '|>' + '
2
+ ' + message['content'] + '<|end|>
3
+ ' }}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>
4
+ ' }}{% else %}{{ eos_token }}{% endif %}
epoch_2/rexmoe_routers.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fc9311a3ddaa16c1f24cf6386064e55ae143c3b934c679cada605eaaf54eece
3
+ size 12619762
epoch_2/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
epoch_2/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
epoch_2/tokenizer_config.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": true,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "32000": {
31
+ "content": "<|endoftext|>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": true
37
+ },
38
+ "32001": {
39
+ "content": "<|assistant|>",
40
+ "lstrip": false,
41
+ "normalized": false,
42
+ "rstrip": true,
43
+ "single_word": false,
44
+ "special": true
45
+ },
46
+ "32002": {
47
+ "content": "<|placeholder1|>",
48
+ "lstrip": false,
49
+ "normalized": false,
50
+ "rstrip": true,
51
+ "single_word": false,
52
+ "special": true
53
+ },
54
+ "32003": {
55
+ "content": "<|placeholder2|>",
56
+ "lstrip": false,
57
+ "normalized": false,
58
+ "rstrip": true,
59
+ "single_word": false,
60
+ "special": true
61
+ },
62
+ "32004": {
63
+ "content": "<|placeholder3|>",
64
+ "lstrip": false,
65
+ "normalized": false,
66
+ "rstrip": true,
67
+ "single_word": false,
68
+ "special": true
69
+ },
70
+ "32005": {
71
+ "content": "<|placeholder4|>",
72
+ "lstrip": false,
73
+ "normalized": false,
74
+ "rstrip": true,
75
+ "single_word": false,
76
+ "special": true
77
+ },
78
+ "32006": {
79
+ "content": "<|system|>",
80
+ "lstrip": false,
81
+ "normalized": false,
82
+ "rstrip": true,
83
+ "single_word": false,
84
+ "special": true
85
+ },
86
+ "32007": {
87
+ "content": "<|end|>",
88
+ "lstrip": false,
89
+ "normalized": false,
90
+ "rstrip": true,
91
+ "single_word": false,
92
+ "special": true
93
+ },
94
+ "32008": {
95
+ "content": "<|placeholder5|>",
96
+ "lstrip": false,
97
+ "normalized": false,
98
+ "rstrip": true,
99
+ "single_word": false,
100
+ "special": true
101
+ },
102
+ "32009": {
103
+ "content": "<|placeholder6|>",
104
+ "lstrip": false,
105
+ "normalized": false,
106
+ "rstrip": true,
107
+ "single_word": false,
108
+ "special": true
109
+ },
110
+ "32010": {
111
+ "content": "<|user|>",
112
+ "lstrip": false,
113
+ "normalized": false,
114
+ "rstrip": true,
115
+ "single_word": false,
116
+ "special": true
117
+ }
118
+ },
119
+ "bos_token": "<s>",
120
+ "clean_up_tokenization_spaces": false,
121
+ "eos_token": "<|endoftext|>",
122
+ "extra_special_tokens": {},
123
+ "legacy": false,
124
+ "model_max_length": 4096,
125
+ "pad_token": "<|endoftext|>",
126
+ "padding_side": "left",
127
+ "sp_model_kwargs": {},
128
+ "tokenizer_class": "LlamaTokenizerFast",
129
+ "unk_token": "<unk>",
130
+ "use_default_system_prompt": false
131
+ }
logs/rexmoe_training_0504_182818.log ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ================================================================================
2
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ReXMoE Training Log - 0504_182818
3
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Log file: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3/logs/rexmoe_training_0504_182818.log
4
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ================================================================================
5
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ================================================================================
6
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ReXMoE Recovery Training (Fine-tuning Pruned Model)
7
+ 2026-04-05 18:28:18 - ReXMoE - INFO - ================================================================================
8
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Checkpoint Path: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
9
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Dataset Path: ../dataset/alpaca_data_cleaned.json
10
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Dataset Mode: IF
11
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Num Samples: 5000
12
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Num Epochs: 1
13
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Batch Size: 1
14
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Max Seq Length: 512
15
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Learning Rate: 0.0001
16
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Save Path: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
17
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Gradient Checkpointing: False
18
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Full LoRA (q,k,v,o): True
19
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Use Scheduler: True
20
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Evaluation Steps: 5000
21
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Gradient Checkpointing: False
22
+ 2026-04-05 18:28:18 - ReXMoE - INFO - Using device: cuda
23
+ 2026-04-05 18:28:18 - ReXMoE - INFO -
24
+ [1/5] Loading pruned model from: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
25
+ 2026-04-05 18:35:31 - ReXMoE - INFO - Loading dense model for KD from microsoft/Phi-mini-MoE-instruct
26
+ 2026-04-05 18:35:35 - ReXMoE - INFO -
27
+ [2/5] Setting up LoRA for fine-tuning...
28
+ 2026-04-05 18:35:36 - ReXMoE - INFO -
29
+ Attempting to load and remap pretrained LoRA from: ./0304_033137_10_rexmoe_natural_phi_mini_moe_R3
30
+ 2026-04-05 18:35:36 - ReXMoE - INFO - ✓ Loaded remapped LoRA weights.
31
+ 2026-04-05 18:35:36 - ReXMoE - INFO - - Remapped expert weights: 0
32
+ 2026-04-05 18:35:36 - ReXMoE - INFO - - Skipped pruned weights: 0
33
+ 2026-04-05 18:35:36 - ReXMoE - INFO - Total parameters: 3,178,713,408
34
+ 2026-04-05 18:35:36 - ReXMoE - INFO - Trainable parameters: 44,938,240 (1.4137%)
35
+ 2026-04-05 18:35:40 - ReXMoE - INFO -
36
+ Prior Evaluation:
37
+ 2026-04-05 18:35:40 - ReXMoE - INFO -
38
+ Evaluating model with 3 sample prompts...
39
+ 2026-04-05 18:36:02 - ReXMoE - INFO -
40
+ --- Prompt 1/3 ---
41
+ 2026-04-05 18:36:02 - ReXMoE - INFO - Instruction: What is the capital of France?
42
+ 2026-04-05 18:36:02 - ReXMoE - INFO - Input: None
43
+ 2026-04-05 18:36:02 - ReXMoE - INFO - Generated completion (len 100): aaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa wasaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaallaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
44
+ 2026-04-05 18:36:23 - ReXMoE - INFO -
45
+ --- Prompt 2/3 ---
46
+ 2026-04-05 18:36:23 - ReXMoE - INFO - Instruction: High-pressure systems stop air from rising into the colder regions of the atmosphere where water can condense. What will most likely result if a high-pressure system remains in an area for a long period of time?
47
+ A. fog
48
+ B. rain
49
+ C. drought
50
+ D. tornado
51
+ Answer:
52
+ 2026-04-05 18:36:23 - ReXMoE - INFO - Input: None
53
+ 2026-04-05 18:36:23 - ReXMoE - INFO - Generated completion (len 100): aaaaaaaaaaaaaallaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa asaaaaaaaaaallaaaaaallaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaallaaaaaaaaaaaa
54
+ aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa0aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
55
+ 2026-04-05 18:36:45 - ReXMoE - INFO -
56
+ --- Prompt 3/3 ---
57
+ 2026-04-05 18:36:45 - ReXMoE - INFO - Instruction: Given the fact: predators eat prey
58
+ Question: Predators eat
59
+ A. lions
60
+ B. humans
61
+ C. bunnies
62
+ D. grass
63
+ Answer:
64
+ 2026-04-05 18:36:45 - ReXMoE - INFO - Input: None
65
+ 2026-04-05 18:36:45 - ReXMoE - INFO - Generated completion (len 100): aall
66
+ aaaaaaaaaaaaaaaaaaaaaaaaaaaallllaaaaaaaaaallaaaa
67
+ aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
68
+ 2026-04-05 18:36:45 - ReXMoE - INFO - Evaluation of all 3 prompts complete.
69
+ 2026-04-05 18:36:45 - ReXMoE - INFO -
70
+ Epoch 1/1
71
+ 2026-04-05 18:51:04 - ReXMoE - INFO - Step 500/5000 | Loss: 3.2554 | KD Loss: 0.9844
72
+ 2026-04-05 19:06:08 - ReXMoE - INFO - Step 1000/5000 | Loss: 2.0955 | KD Loss: 1.5000
73
+ 2026-04-05 19:21:21 - ReXMoE - INFO - Step 1500/5000 | Loss: 2.2041 | KD Loss: 1.0781
74
+ 2026-04-05 19:36:38 - ReXMoE - INFO - Step 2000/5000 | Loss: 3.4660 | KD Loss: 1.4219
75
+ 2026-04-05 19:51:48 - ReXMoE - INFO - Step 2500/5000 | Loss: 1.8320 | KD Loss: 0.8945
76
+ 2026-04-05 20:07:07 - ReXMoE - INFO - Step 3000/5000 | Loss: 2.4157 | KD Loss: 1.4219
77
+ 2026-04-05 20:22:22 - ReXMoE - INFO - Step 3500/5000 | Loss: 1.9184 | KD Loss: 1.3516
78
+ 2026-04-05 20:37:35 - ReXMoE - INFO - Step 4000/5000 | Loss: 1.4622 | KD Loss: 0.7266
79
+ 2026-04-05 20:52:49 - ReXMoE - INFO - Step 4500/5000 | Loss: 1.9962 | KD Loss: 1.2344
80
+ 2026-04-05 21:08:04 - ReXMoE - INFO - Step 5000/5000 | Loss: 1.7498 | KD Loss: 1.0000
81
+ 2026-04-05 21:08:04 - ReXMoE - INFO -
82
+ Evaluation at step 5000:
83
+ 2026-04-05 21:08:04 - ReXMoE - INFO -
84
+ Evaluating model with 3 sample prompts...
85
+ 2026-04-05 21:08:06 - ReXMoE - INFO -
86
+ --- Prompt 1/3 ---
87
+ 2026-04-05 21:08:06 - ReXMoE - INFO - Instruction: What is the capital of France?
88
+ 2026-04-05 21:08:06 - ReXMoE - INFO - Input: None
89
+ 2026-04-05 21:08:06 - ReXMoE - INFO - Generated completion (len 9): The capital of France is Paris.
90
+ 2026-04-05 21:08:12 - ReXMoE - INFO -
91
+ --- Prompt 2/3 ---
92
+ 2026-04-05 21:08:12 - ReXMoE - INFO - Instruction: High-pressure systems stop air from rising into the colder regions of the atmosphere where water can condense. What will most likely result if a high-pressure system remains in an area for a long period of time?
93
+ A. fog
94
+ B. rain
95
+ C. drought
96
+ D. tornado
97
+ Answer:
98
+ 2026-04-05 21:08:12 - ReXMoE - INFO - Input: None
99
+ 2026-04-05 21:08:12 - ReXMoE - INFO - Generated completion (len 25): The most likely result of a high-pressure system working up for an extended period of time is a fog.
100
+ 2026-04-05 21:08:17 - ReXMoE - INFO -
101
+ --- Prompt 3/3 ---
102
+ 2026-04-05 21:08:17 - ReXMoE - INFO - Instruction: Given the fact: predators eat prey
103
+ Question: Predators eat
104
+ A. lions
105
+ B. humans
106
+ C. bunnies
107
+ D. grass
108
+ Answer:
109
+ 2026-04-05 21:08:17 - ReXMoE - INFO - Input: None
110
+ 2026-04-05 21:08:17 - ReXMoE - INFO - Generated completion (len 21): A. lions
111
+ B. humans
112
+ C. bunnies
113
+ D. grass.
114
+ 2026-04-05 21:08:17 - ReXMoE - INFO - Evaluation of all 3 prompts complete.
115
+ 2026-04-05 21:08:17 - ReXMoE - INFO - Epoch 1 completed. Average Loss: nan
116
+ 2026-04-05 21:08:21 - ReXMoE - INFO - Training complete.
logs/rexmoe_training_0604_042639.log ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ================================================================================
2
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ReXMoE Training Log - 0604_042639
3
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Log file: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3/logs/rexmoe_training_0604_042639.log
4
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ================================================================================
5
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ================================================================================
6
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ReXMoE Recovery Training (Fine-tuning Pruned Model)
7
+ 2026-04-06 04:26:39 - ReXMoE - INFO - ================================================================================
8
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Checkpoint Path: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
9
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Dataset Path: ../dataset/alpaca_data_cleaned.json
10
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Dataset Mode: IF
11
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Num Samples: 10000
12
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Num Epochs: 1
13
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Batch Size: 1
14
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Max Seq Length: 512
15
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Learning Rate: 0.0001
16
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Save Path: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
17
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Gradient Checkpointing: False
18
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Full LoRA (q,k,v,o): True
19
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Use Scheduler: True
20
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Evaluation Steps: 10000
21
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Gradient Checkpointing: False
22
+ 2026-04-06 04:26:39 - ReXMoE - INFO - Using device: cuda
23
+ 2026-04-06 04:26:39 - ReXMoE - INFO -
24
+ [1/5] Loading pruned model from: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
logs/rexmoe_training_0604_043401.log ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ================================================================================
2
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ReXMoE Training Log - 0604_043401
3
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Log file: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3/logs/rexmoe_training_0604_043401.log
4
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ================================================================================
5
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ================================================================================
6
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ReXMoE Recovery Training (Fine-tuning Pruned Model)
7
+ 2026-04-06 04:34:01 - ReXMoE - INFO - ================================================================================
8
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Checkpoint Path: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
9
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Dataset Path: ../dataset/alpaca_data_cleaned.json
10
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Dataset Mode: IF_2
11
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Num Samples: 500
12
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Num Epochs: 2
13
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Batch Size: 1
14
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Max Seq Length: 512
15
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Learning Rate: 0.0001
16
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Save Path: ./rexmoe_recovered_75_pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
17
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Gradient Checkpointing: False
18
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Full LoRA (q,k,v,o): True
19
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Use Scheduler: True
20
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Evaluation Steps: 10000
21
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Gradient Checkpointing: False
22
+ 2026-04-06 04:34:01 - ReXMoE - INFO - Using device: cuda
23
+ 2026-04-06 04:34:01 - ReXMoE - INFO -
24
+ [1/5] Loading pruned model from: ./pruned_75_0304_033137_10_rexmoe_natural_phi_mini_moe_R3
25
+ 2026-04-06 04:41:17 - ReXMoE - INFO - Loading dense model for KD from microsoft/Phi-mini-MoE-instruct
26
+ 2026-04-06 04:41:22 - ReXMoE - INFO -
27
+ [2/5] Setting up LoRA for fine-tuning...
28
+ 2026-04-06 04:41:23 - ReXMoE - INFO -
29
+ Attempting to load and remap pretrained LoRA from: ./0304_033137_10_rexmoe_natural_phi_mini_moe_R3
30
+ 2026-04-06 04:41:23 - ReXMoE - INFO - ✓ Loaded remapped LoRA weights.
31
+ 2026-04-06 04:41:23 - ReXMoE - INFO - - Remapped expert weights: 0
32
+ 2026-04-06 04:41:23 - ReXMoE - INFO - - Skipped pruned weights: 0
33
+ 2026-04-06 04:41:23 - ReXMoE - INFO - Total parameters: 3,178,713,408
34
+ 2026-04-06 04:41:23 - ReXMoE - INFO - Trainable parameters: 44,938,240 (1.4137%)
35
+ 2026-04-06 04:41:26 - ReXMoE - INFO -
36
+ Prior Evaluation:
37
+ 2026-04-06 04:41:26 - ReXMoE - INFO -
38
+ Evaluating model with 3 sample prompts...
39
+ 2026-04-06 04:41:51 - ReXMoE - INFO -
40
+ --- Prompt 1/3 ---
41
+ 2026-04-06 04:41:51 - ReXMoE - INFO - Instruction: What is the capital of France?
42
+ 2026-04-06 04:41:51 - ReXMoE - INFO - Input: None
43
+ 2026-04-06 04:41:51 - ReXMoE - INFO - Generated completion (len 100): aaaaaaaaaaaaaaaaaaaaaaaaaa aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa!aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaallaaaaaaaaaaaa
44
+ 2026-04-06 04:42:14 - ReXMoE - INFO -
45
+ --- Prompt 2/3 ---
46
+ 2026-04-06 04:42:14 - ReXMoE - INFO - Instruction: High-pressure systems stop air from rising into the colder regions of the atmosphere where water can condense. What will most likely result if a high-pressure system remains in an area for a long period of time?
47
+ A. fog
48
+ B. rain
49
+ C. drought
50
+ D. tornado
51
+ Answer:
52
+ 2026-04-06 04:42:14 - ReXMoE - INFO - Input: None
53
+ 2026-04-06 04:42:14 - ReXMoE - INFO - Generated completion (len 100): aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa1aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
54
+ 2026-04-06 04:42:38 - ReXMoE - INFO -
55
+ --- Prompt 3/3 ---
56
+ 2026-04-06 04:42:38 - ReXMoE - INFO - Instruction: Given the fact: predators eat prey
57
+ Question: Predators eat
58
+ A. lions
59
+ B. humans
60
+ C. bunnies
61
+ D. grass
62
+ Answer:
63
+ 2026-04-06 04:42:38 - ReXMoE - INFO - Input: None
64
+ 2026-04-06 04:42:38 - ReXMoE - INFO - Generated completion (len 100): aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa_aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
65
+ 2026-04-06 04:42:38 - ReXMoE - INFO - Evaluation of all 3 prompts complete.
66
+ 2026-04-06 04:42:38 - ReXMoE - INFO -
67
+ Epoch 1/2
68
+ 2026-04-06 04:45:42 - ReXMoE - INFO - Step 100/1000 | Loss: 4.0626 | KD Loss: 1.4453
69
+ 2026-04-06 04:49:01 - ReXMoE - INFO - Step 200/1000 | Loss: 4.2264 | KD Loss: 3.7812
70
+ 2026-04-06 04:52:20 - ReXMoE - INFO - Step 300/1000 | Loss: 3.6379 | KD Loss: 2.4844
71
+ 2026-04-06 04:55:38 - ReXMoE - INFO - Step 400/1000 | Loss: 3.0670 | KD Loss: 1.7109
72
+ 2026-04-06 04:58:58 - ReXMoE - INFO - Step 500/1000 | Loss: nan | KD Loss: 2.1875
73
+ 2026-04-06 04:58:58 - ReXMoE - INFO - Epoch 1 completed. Average Loss: nan
74
+ 2026-04-06 04:59:04 - ReXMoE - INFO -
75
+ Epoch 2/2
76
+ 2026-04-06 05:02:25 - ReXMoE - INFO - Step 600/1000 | Loss: 2.6729 | KD Loss: 1.8906
77
+ 2026-04-06 05:05:45 - ReXMoE - INFO - Step 700/1000 | Loss: 1.6074 | KD Loss: 1.2188
78
+ 2026-04-06 05:09:07 - ReXMoE - INFO - Step 800/1000 | Loss: 1.8340 | KD Loss: 1.2812
79
+ 2026-04-06 05:12:29 - ReXMoE - INFO - Step 900/1000 | Loss: 2.6532 | KD Loss: 1.2891
80
+ 2026-04-06 05:15:50 - ReXMoE - INFO - Step 1000/1000 | Loss: 1.7152 | KD Loss: 1.1719
81
+ 2026-04-06 05:15:50 - ReXMoE - INFO - Epoch 2 completed. Average Loss: nan
82
+ 2026-04-06 05:15:54 - ReXMoE - INFO - Training complete.