AA65327 commited on
Commit
7c161ee
·
verified ·
1 Parent(s): 049a5fc

Upload fine-tuned LLaMA-2-7B emotion analysis model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ 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
+ hallucination_analysis.png filter=lfs diff=lfs merge=lfs -text
37
+ performance_metrics.png filter=lfs diff=lfs merge=lfs -text
38
+ training_curves.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,124 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: NousResearch/Llama-2-7b-chat-hf
4
+ tags:
5
+ - llama2
6
+ - emotion-analysis
7
+ - activity-context
8
+ - fine-tuned
9
+ - goemotions
10
+ datasets:
11
+ - AA65327/GoEmotions_Alpaca_Final
12
+ language:
13
+ - en
14
+ pipeline_tag: text-generation
15
+ ---
16
+
17
+ # LLaMA-2-7B Emotion Analysis with Activity Context
18
+
19
+ ## Model Description
20
+
21
+ This model is a fine-tuned version of NousResearch/Llama-2-7b-chat-hf on the GoEmotions dataset with activity context integration.
22
+ It analyzes emotions in text while considering the user's recent activity patterns to provide more contextual insights.
23
+
24
+ ## Training Details
25
+
26
+ ### Training Data
27
+ - **Dataset**: AA65327/GoEmotions_Alpaca_Final
28
+ - **Training samples**: N/A
29
+ - **Validation samples**: N/A
30
+
31
+ ### Training Configuration
32
+ - **Base model**: NousResearch/Llama-2-7b-chat-hf
33
+ - **Training epochs**: 1
34
+ - **Batch size**: 1
35
+ - **Learning rate**: 0.0002
36
+ - **LoRA rank**: 8
37
+ - **LoRA alpha**: 32
38
+
39
+ ## Performance Metrics
40
+
41
+ ### Evaluation Results
42
+ - **Perplexity**: 26.08
43
+ - **ROUGE-1**: 0.190
44
+ - **ROUGE-2**: 0.170
45
+ - **ROUGE-L**: 0.190
46
+ - **BLEU Score**: 8.039
47
+ - **Inference Speed**: 1.3 tokens/sec
48
+ - **Hallucination Rate**: 2.400
49
+
50
+ ## Usage
51
+
52
+ ```python
53
+ from transformers import AutoTokenizer, AutoModelForCausalLM
54
+ from peft import PeftModel
55
+
56
+ # Load model and tokenizer
57
+ base_model = AutoModelForCausalLM.from_pretrained(
58
+ "NousResearch/Llama-2-7b-chat-hf",
59
+ load_in_4bit=True,
60
+ device_map="auto"
61
+ )
62
+ model = PeftModel.from_pretrained(base_model, "AA65327/llama2-emotion-activity-20251005")
63
+ tokenizer = AutoTokenizer.from_pretrained("AA65327/llama2-emotion-activity-20251005")
64
+
65
+ # Format your prompt
66
+ def format_prompt(instruction, input_text, activity_log):
67
+ return f"""Below is an instruction that describes a task, paired with an input that provides further context.
68
+ Write a response that appropriately completes the request.
69
+
70
+ ### Instruction:
71
+ {instruction}
72
+
73
+ ### Input:
74
+ Current message: {input_text}
75
+ Activity log (past 3 days, hours per activity): {activity_log}
76
+
77
+ ### Response:
78
+ """
79
+
80
+ # Example usage
81
+ instruction = "Evaluate the emotion in this text and suggest why the person might feel this way."
82
+ input_text = "I'm feeling really excited about this new project!"
83
+ activity_log = "working_out: [2, 1, 3]; reading: [1, 2, 0]; socializing: [3, 4, 2]"
84
+
85
+ prompt = format_prompt(instruction, input_text, activity_log)
86
+ inputs = tokenizer(prompt, return_tensors="pt")
87
+ outputs = model.generate(**inputs, max_new_tokens=150, temperature=0.7)
88
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
89
+ print(response)
90
+ ```
91
+
92
+ ## Training Procedure
93
+
94
+ The model was trained using LoRA (Low-Rank Adaptation) technique with the following approach:
95
+ 1. Load base LLaMA-2-7B-Chat model with 4-bit quantization
96
+ 2. Apply LoRA adapters to query and value projection layers
97
+ 3. Fine-tune on emotion analysis tasks with activity context
98
+ 4. Implement gradient checkpointing and mixed precision training
99
+ 5. Use early stopping based on validation loss
100
+
101
+ ## Limitations and Bias
102
+
103
+ - The model may reflect biases present in the training data
104
+ - Performance may vary on domains not represented in the training set
105
+ - Activity context interpretation is based on patterns learned from training data
106
+ - Generated content should be reviewed for factual accuracy
107
+
108
+ ## Citation
109
+
110
+ ```bibtex
111
+ @misc{llama2-emotion-activity-2025,
112
+ author = {AA65327},
113
+ title = {LLaMA-2-7B Emotion Analysis with Activity Context},
114
+ year = {2025},
115
+ publisher = {Hugging Face},
116
+ url = {https://huggingface.co/AA65327/llama2-emotion-activity-20251005}
117
+ }
118
+ ```
119
+
120
+ ## Acknowledgments
121
+
122
+ - Meta AI for the base LLaMA-2 model
123
+ - Google Research for the GoEmotions dataset
124
+ - Hugging Face for the transformers library and model hosting
adapter_config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "NousResearch/Llama-2-7b-chat-hf",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 8,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "q_proj",
29
+ "k_proj",
30
+ "v_proj",
31
+ "o_proj"
32
+ ],
33
+ "target_parameters": null,
34
+ "task_type": "CAUSAL_LM",
35
+ "trainable_token_indices": null,
36
+ "use_dora": false,
37
+ "use_qalora": false,
38
+ "use_rslora": false
39
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2898596731c6d103dac68e0c797a69bd04469c5066cf458134301caa68597dde
3
+ size 33588528
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<pad>": 32000
3
+ }
checkpoint-50/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: NousResearch/Llama-2-7b-chat-hf
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:NousResearch/Llama-2-7b-chat-hf
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.17.1
checkpoint-50/adapter_config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "NousResearch/Llama-2-7b-chat-hf",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 8,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "q_proj",
29
+ "k_proj",
30
+ "v_proj",
31
+ "o_proj"
32
+ ],
33
+ "target_parameters": null,
34
+ "task_type": "CAUSAL_LM",
35
+ "trainable_token_indices": null,
36
+ "use_dora": false,
37
+ "use_qalora": false,
38
+ "use_rslora": false
39
+ }
checkpoint-50/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2898596731c6d103dac68e0c797a69bd04469c5066cf458134301caa68597dde
3
+ size 33588528
checkpoint-50/added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<pad>": 32000
3
+ }
checkpoint-50/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f5d92048df8e9c44ffe67116a14424df38bb672c781b15b009c9c28d65b2d63
3
+ size 67327691
checkpoint-50/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d073bd6c91f8e22d16ac1a0c8d491664eb3be2f8b609f9f44458561f061786ca
3
+ size 14645
checkpoint-50/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4293d1449e7b8ded03fb6577e4185f63efc99517fd5f93bfced72522213551b2
3
+ size 1383
checkpoint-50/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31edc2a2bbce91c1e6afed681e5dc14966c375f87f735113b40ed0fe4d96a219
3
+ size 1465
checkpoint-50/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<unk>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
checkpoint-50/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-50/tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
checkpoint-50/tokenizer_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": true,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": true,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ },
30
+ "32000": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ }
38
+ },
39
+ "bos_token": "<s>",
40
+ "clean_up_tokenization_spaces": false,
41
+ "eos_token": "</s>",
42
+ "extra_special_tokens": {},
43
+ "legacy": false,
44
+ "model_max_length": 1000000000000000019884624838656,
45
+ "pad_token": "<unk>",
46
+ "sp_model_kwargs": {},
47
+ "tokenizer_class": "LlamaTokenizer",
48
+ "unk_token": "<unk>",
49
+ "use_default_system_prompt": false
50
+ }
checkpoint-50/trainer_state.json ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 50,
3
+ "best_metric": 0.9238768815994263,
4
+ "best_model_checkpoint": "./llama2-goemotions-lora-optimized/checkpoint-50",
5
+ "epoch": 0.001151808339092375,
6
+ "eval_steps": 25,
7
+ "global_step": 50,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.000230361667818475,
14
+ "grad_norm": 2.6401069164276123,
15
+ "learning_rate": 0.00018,
16
+ "loss": 3.5448,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.00046072333563695,
21
+ "grad_norm": 3.533531904220581,
22
+ "learning_rate": 0.000155,
23
+ "loss": 2.0609,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.0005759041695461875,
28
+ "eval_loss": 1.1123775243759155,
29
+ "eval_runtime": 6017.5045,
30
+ "eval_samples_per_second": 1.804,
31
+ "eval_steps_per_second": 1.804,
32
+ "step": 25
33
+ },
34
+ {
35
+ "epoch": 0.000691085003455425,
36
+ "grad_norm": 1.6841713190078735,
37
+ "learning_rate": 0.000105,
38
+ "loss": 1.1921,
39
+ "step": 30
40
+ },
41
+ {
42
+ "epoch": 0.0009214466712739,
43
+ "grad_norm": 1.636803388595581,
44
+ "learning_rate": 5.500000000000001e-05,
45
+ "loss": 1.0012,
46
+ "step": 40
47
+ },
48
+ {
49
+ "epoch": 0.001151808339092375,
50
+ "grad_norm": 1.3772783279418945,
51
+ "learning_rate": 5e-06,
52
+ "loss": 0.9673,
53
+ "step": 50
54
+ },
55
+ {
56
+ "epoch": 0.001151808339092375,
57
+ "eval_loss": 0.9238768815994263,
58
+ "eval_runtime": 6019.9295,
59
+ "eval_samples_per_second": 1.803,
60
+ "eval_steps_per_second": 1.803,
61
+ "step": 50
62
+ }
63
+ ],
64
+ "logging_steps": 10,
65
+ "max_steps": 50,
66
+ "num_input_tokens_seen": 0,
67
+ "num_train_epochs": 1,
68
+ "save_steps": 50,
69
+ "stateful_callbacks": {
70
+ "TrainerControl": {
71
+ "args": {
72
+ "should_epoch_stop": false,
73
+ "should_evaluate": false,
74
+ "should_log": false,
75
+ "should_save": true,
76
+ "should_training_stop": true
77
+ },
78
+ "attributes": {}
79
+ }
80
+ },
81
+ "total_flos": 1016176469606400.0,
82
+ "train_batch_size": 1,
83
+ "trial_name": null,
84
+ "trial_params": null
85
+ }
checkpoint-50/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d041afb86be4a082acdec15891d0f748eda550dea5a4ae87ad52f771ca6ac082
3
+ size 5777
evaluation_metrics.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "perplexity": 26.080894470214844,
3
+ "tokens_per_second": 1.3229382650688788,
4
+ "samples_per_second": 0.11811948795257846,
5
+ "avg_time_per_sample": 8.466003513336181,
6
+ "rouge1": 0.190389919784236,
7
+ "rouge2": 0.16995296818860808,
8
+ "rougeL": 0.190389919784236,
9
+ "bleu": 8.038511090894387,
10
+ "avg_hallucination": 2.40048841144491,
11
+ "hallucination_std": 0.3121710792145274
12
+ }
hallucination_analysis.png ADDED

Git LFS Details

  • SHA256: 53ae63aef5ee57c7b90c710849a4297cd7010704c691c1ae1369e3186e97a290
  • Pointer size: 131 Bytes
  • Size of remote file: 106 kB
metrics.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ perplexity,tokens_per_second,samples_per_second,avg_time_per_sample,rouge1,rouge2,rougeL,bleu,avg_hallucination,hallucination_std
2
+ 26.080894470214844,1.3229382650688788,0.11811948795257846,8.466003513336181,0.190389919784236,0.16995296818860808,0.190389919784236,8.038511090894387,2.40048841144491,0.3121710792145274
performance_metrics.png ADDED

Git LFS Details

  • SHA256: 0089e3e2299801b26c1dea8a871d0ec11e920d3c5d212999eef4b0681cdf5539
  • Pointer size: 131 Bytes
  • Size of remote file: 181 kB
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<unk>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
tokenizer_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": true,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": true,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ },
30
+ "32000": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ }
38
+ },
39
+ "bos_token": "<s>",
40
+ "clean_up_tokenization_spaces": false,
41
+ "eos_token": "</s>",
42
+ "extra_special_tokens": {},
43
+ "legacy": false,
44
+ "model_max_length": 1000000000000000019884624838656,
45
+ "pad_token": "<unk>",
46
+ "sp_model_kwargs": {},
47
+ "tokenizer_class": "LlamaTokenizer",
48
+ "unk_token": "<unk>",
49
+ "use_default_system_prompt": false
50
+ }
trainer_state.json ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 50,
3
+ "best_metric": 0.9238768815994263,
4
+ "best_model_checkpoint": "./llama2-goemotions-lora-optimized/checkpoint-50",
5
+ "epoch": 0.001151808339092375,
6
+ "eval_steps": 25,
7
+ "global_step": 50,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.000230361667818475,
14
+ "grad_norm": 2.6401069164276123,
15
+ "learning_rate": 0.00018,
16
+ "loss": 3.5448,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.00046072333563695,
21
+ "grad_norm": 3.533531904220581,
22
+ "learning_rate": 0.000155,
23
+ "loss": 2.0609,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.0005759041695461875,
28
+ "eval_loss": 1.1123775243759155,
29
+ "eval_runtime": 6017.5045,
30
+ "eval_samples_per_second": 1.804,
31
+ "eval_steps_per_second": 1.804,
32
+ "step": 25
33
+ },
34
+ {
35
+ "epoch": 0.000691085003455425,
36
+ "grad_norm": 1.6841713190078735,
37
+ "learning_rate": 0.000105,
38
+ "loss": 1.1921,
39
+ "step": 30
40
+ },
41
+ {
42
+ "epoch": 0.0009214466712739,
43
+ "grad_norm": 1.636803388595581,
44
+ "learning_rate": 5.500000000000001e-05,
45
+ "loss": 1.0012,
46
+ "step": 40
47
+ },
48
+ {
49
+ "epoch": 0.001151808339092375,
50
+ "grad_norm": 1.3772783279418945,
51
+ "learning_rate": 5e-06,
52
+ "loss": 0.9673,
53
+ "step": 50
54
+ },
55
+ {
56
+ "epoch": 0.001151808339092375,
57
+ "eval_loss": 0.9238768815994263,
58
+ "eval_runtime": 6019.9295,
59
+ "eval_samples_per_second": 1.803,
60
+ "eval_steps_per_second": 1.803,
61
+ "step": 50
62
+ },
63
+ {
64
+ "epoch": 0.001151808339092375,
65
+ "step": 50,
66
+ "total_flos": 1016176469606400.0,
67
+ "train_loss": 1.7532618713378907,
68
+ "train_runtime": 12121.5826,
69
+ "train_samples_per_second": 0.004,
70
+ "train_steps_per_second": 0.004
71
+ }
72
+ ],
73
+ "logging_steps": 10,
74
+ "max_steps": 50,
75
+ "num_input_tokens_seen": 0,
76
+ "num_train_epochs": 1,
77
+ "save_steps": 50,
78
+ "stateful_callbacks": {
79
+ "TrainerControl": {
80
+ "args": {
81
+ "should_epoch_stop": false,
82
+ "should_evaluate": false,
83
+ "should_log": false,
84
+ "should_save": true,
85
+ "should_training_stop": true
86
+ },
87
+ "attributes": {}
88
+ }
89
+ },
90
+ "total_flos": 1016176469606400.0,
91
+ "train_batch_size": 1,
92
+ "trial_name": null,
93
+ "trial_params": null
94
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d041afb86be4a082acdec15891d0f748eda550dea5a4ae87ad52f771ca6ac082
3
+ size 5777
training_curves.png ADDED

Git LFS Details

  • SHA256: 10ccbc2c1d504626c2afd0ce0858ed8db1d5b5465c0695ce9c6924037060d7cc
  • Pointer size: 131 Bytes
  • Size of remote file: 192 kB