diaenra commited on
Commit
66f92dc
1 Parent(s): 1fe8e01

End of training

Browse files
Files changed (2) hide show
  1. README.md +148 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,148 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: princeton-nlp/Sheared-LLaMA-1.3B
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: 837353c6-bfd2-4a87-979d-591305107c94
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
17
+ <details><summary>See axolotl config</summary>
18
+
19
+ axolotl version: `0.4.1`
20
+ ```yaml
21
+ adapter: lora
22
+ base_model: princeton-nlp/Sheared-LLaMA-1.3B
23
+ bf16: auto
24
+ chat_template: llama3
25
+ dataset_prepared_path: null
26
+ datasets:
27
+ - data_files:
28
+ - d4aaa754d4db65bf_train_data.json
29
+ ds_type: json
30
+ field: text
31
+ path: /workspace/input_data/d4aaa754d4db65bf_train_data.json
32
+ type: completion
33
+ debug: null
34
+ deepspeed: null
35
+ early_stopping_patience: 1
36
+ eval_max_new_tokens: 128
37
+ eval_steps: 25
38
+ eval_table_size: null
39
+ flash_attention: true
40
+ fp16: false
41
+ fsdp: null
42
+ fsdp_config: null
43
+ gradient_accumulation_steps: 4
44
+ gradient_checkpointing: true
45
+ group_by_length: true
46
+ hub_model_id: diaenra/837353c6-bfd2-4a87-979d-591305107c94
47
+ hub_repo: null
48
+ hub_strategy: checkpoint
49
+ hub_token: null
50
+ learning_rate: 0.0002
51
+ load_in_4bit: false
52
+ load_in_8bit: false
53
+ local_rank: null
54
+ logging_steps: 1
55
+ lora_alpha: 32
56
+ lora_dropout: 0.05
57
+ lora_fan_in_fan_out: null
58
+ lora_model_dir: null
59
+ lora_r: 16
60
+ lora_target_linear: true
61
+ lr_scheduler: cosine
62
+ max_steps: 100
63
+ micro_batch_size: 2
64
+ mlflow_experiment_name: /tmp/d4aaa754d4db65bf_train_data.json
65
+ model_type: AutoModelForCausalLM
66
+ num_epochs: 1
67
+ optimizer: adamw_torch
68
+ output_dir: miner_id_24
69
+ pad_to_sequence_len: true
70
+ resume_from_checkpoint: null
71
+ s2_attention: null
72
+ sample_packing: false
73
+ save_steps: 25
74
+ sequence_len: 4056
75
+ special_tokens:
76
+ pad_token: </s>
77
+ strict: false
78
+ tf32: false
79
+ tokenizer_type: AutoTokenizer
80
+ train_on_inputs: false
81
+ trust_remote_code: true
82
+ val_set_size: 0.05
83
+ wandb_entity: diaenra-tao-miner
84
+ wandb_mode: online
85
+ wandb_name: 837353c6-bfd2-4a87-979d-591305107c94
86
+ wandb_project: tao
87
+ wandb_run: diaenra
88
+ wandb_runid: 837353c6-bfd2-4a87-979d-591305107c94
89
+ warmup_ratio: 0.05
90
+ weight_decay: 0.01
91
+ xformers_attention: true
92
+
93
+ ```
94
+
95
+ </details><br>
96
+
97
+ # 837353c6-bfd2-4a87-979d-591305107c94
98
+
99
+ This model is a fine-tuned version of [princeton-nlp/Sheared-LLaMA-1.3B](https://huggingface.co/princeton-nlp/Sheared-LLaMA-1.3B) on the None dataset.
100
+ It achieves the following results on the evaluation set:
101
+ - Loss: 2.6855
102
+
103
+ ## Model description
104
+
105
+ More information needed
106
+
107
+ ## Intended uses & limitations
108
+
109
+ More information needed
110
+
111
+ ## Training and evaluation data
112
+
113
+ More information needed
114
+
115
+ ## Training procedure
116
+
117
+ ### Training hyperparameters
118
+
119
+ The following hyperparameters were used during training:
120
+ - learning_rate: 0.0002
121
+ - train_batch_size: 2
122
+ - eval_batch_size: 2
123
+ - seed: 42
124
+ - gradient_accumulation_steps: 4
125
+ - total_train_batch_size: 8
126
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
127
+ - lr_scheduler_type: cosine
128
+ - lr_scheduler_warmup_steps: 5
129
+ - training_steps: 100
130
+
131
+ ### Training results
132
+
133
+ | Training Loss | Epoch | Step | Validation Loss |
134
+ |:-------------:|:------:|:----:|:---------------:|
135
+ | 2.2693 | 0.0001 | 1 | 2.8410 |
136
+ | 2.5126 | 0.0037 | 25 | 2.7154 |
137
+ | 2.7793 | 0.0074 | 50 | 2.7025 |
138
+ | 2.7395 | 0.0111 | 75 | 2.6895 |
139
+ | 2.9277 | 0.0148 | 100 | 2.6855 |
140
+
141
+
142
+ ### Framework versions
143
+
144
+ - PEFT 0.13.2
145
+ - Transformers 4.46.0
146
+ - Pytorch 2.5.0+cu124
147
+ - Datasets 3.0.1
148
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10961fdeb35d2b2f75c83bc67034ba86add90d80c6ff49c521cc668dda01b781
3
+ size 60086346