diaenra commited on
Commit
de35a1f
1 Parent(s): 50dd6ee

End of training

Browse files
Files changed (3) hide show
  1. README.md +153 -0
  2. adapter_model.bin +3 -0
  3. adapter_model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: llama3
4
+ base_model: tokyotech-llm/Llama-3-Swallow-8B-v0.1
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: 8cad90b9-d437-42b4-afb4-e51b67e64e50
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: tokyotech-llm/Llama-3-Swallow-8B-v0.1
23
+ bf16: auto
24
+ chat_template: llama3
25
+ dataset_prepared_path: null
26
+ datasets:
27
+ - data_files:
28
+ - 3f72bc6633b04dfb_train_data.json
29
+ ds_type: json
30
+ format: custom
31
+ path: /workspace/input_data/3f72bc6633b04dfb_train_data.json
32
+ type:
33
+ field_input: titles
34
+ field_instruction: instructions
35
+ field_output: essays
36
+ format: '{instruction} {input}'
37
+ no_input_format: '{instruction}'
38
+ system_format: '{system}'
39
+ system_prompt: ''
40
+ debug: null
41
+ deepspeed: null
42
+ early_stopping_patience: 1
43
+ eval_max_new_tokens: 128
44
+ eval_steps: 25
45
+ eval_table_size: null
46
+ flash_attention: true
47
+ fp16: false
48
+ fsdp: null
49
+ fsdp_config: null
50
+ gradient_accumulation_steps: 4
51
+ gradient_checkpointing: true
52
+ group_by_length: true
53
+ hub_model_id: diaenra/8cad90b9-d437-42b4-afb4-e51b67e64e50
54
+ hub_repo: null
55
+ hub_strategy: checkpoint
56
+ hub_token: null
57
+ learning_rate: 0.0001
58
+ load_in_4bit: false
59
+ load_in_8bit: false
60
+ local_rank: null
61
+ logging_steps: 1
62
+ lora_alpha: 32
63
+ lora_dropout: 0.05
64
+ lora_fan_in_fan_out: null
65
+ lora_model_dir: null
66
+ lora_r: 16
67
+ lora_target_linear: true
68
+ lr_scheduler: cosine
69
+ max_steps: 100
70
+ micro_batch_size: 2
71
+ mlflow_experiment_name: /tmp/3f72bc6633b04dfb_train_data.json
72
+ model_type: AutoModelForCausalLM
73
+ num_epochs: 1
74
+ optimizer: adamw_torch
75
+ output_dir: miner_id_24
76
+ pad_to_sequence_len: true
77
+ resume_from_checkpoint: null
78
+ s2_attention: null
79
+ sample_packing: false
80
+ save_steps: 25
81
+ sequence_len: 2048
82
+ special_tokens:
83
+ pad_token: <|end_of_text|>
84
+ strict: false
85
+ tf32: false
86
+ tokenizer_type: AutoTokenizer
87
+ train_on_inputs: false
88
+ trust_remote_code: true
89
+ val_set_size: 0.05
90
+ wandb_entity: diaenra-tao-miner
91
+ wandb_mode: online
92
+ wandb_name: 8cad90b9-d437-42b4-afb4-e51b67e64e50
93
+ wandb_project: tao
94
+ wandb_run: diaenra
95
+ wandb_runid: 8cad90b9-d437-42b4-afb4-e51b67e64e50
96
+ warmup_ratio: 0.05
97
+ weight_decay: 0.01
98
+ xformers_attention: true
99
+
100
+ ```
101
+
102
+ </details><br>
103
+
104
+ # 8cad90b9-d437-42b4-afb4-e51b67e64e50
105
+
106
+ This model is a fine-tuned version of [tokyotech-llm/Llama-3-Swallow-8B-v0.1](https://huggingface.co/tokyotech-llm/Llama-3-Swallow-8B-v0.1) on the None dataset.
107
+ It achieves the following results on the evaluation set:
108
+ - Loss: 2.1202
109
+
110
+ ## Model description
111
+
112
+ More information needed
113
+
114
+ ## Intended uses & limitations
115
+
116
+ More information needed
117
+
118
+ ## Training and evaluation data
119
+
120
+ More information needed
121
+
122
+ ## Training procedure
123
+
124
+ ### Training hyperparameters
125
+
126
+ The following hyperparameters were used during training:
127
+ - learning_rate: 0.0001
128
+ - train_batch_size: 2
129
+ - eval_batch_size: 2
130
+ - seed: 42
131
+ - gradient_accumulation_steps: 4
132
+ - total_train_batch_size: 8
133
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
134
+ - lr_scheduler_type: cosine
135
+ - lr_scheduler_warmup_steps: 5
136
+ - training_steps: 100
137
+
138
+ ### Training results
139
+
140
+ | Training Loss | Epoch | Step | Validation Loss |
141
+ |:-------------:|:------:|:----:|:---------------:|
142
+ | 2.3432 | 0.0054 | 1 | 2.1719 |
143
+ | 2.2239 | 0.1351 | 25 | 2.1194 |
144
+ | 2.3098 | 0.2703 | 50 | 2.1202 |
145
+
146
+
147
+ ### Framework versions
148
+
149
+ - PEFT 0.13.2
150
+ - Transformers 4.46.0
151
+ - Pytorch 2.5.0+cu124
152
+ - Datasets 3.0.1
153
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3444c9d8ac891bcec39acec3266b4072847da050da155e9172393943446abab8
3
+ size 167934026
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c7715e1ca330a24f251bbcde8e96bfd94ed504aa5bcc04838e50a48cabdb1e0
3
  size 167832240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07f56c5da7f7f4593886a3c0e2c5bbceee50851a16b8bf7e9634876158583518
3
  size 167832240