ncbateman commited on
Commit
cea75f2
·
verified ·
1 Parent(s): 1b7bf54

End of training

Browse files
Files changed (2) hide show
  1. README.md +150 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: EleutherAI/pythia-14m
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: 8f97fcaa-a2e2-4f79-8ff1-0ca9a91cd08f
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ [<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)
16
+ <details><summary>See axolotl config</summary>
17
+
18
+ axolotl version: `0.4.1`
19
+ ```yaml
20
+ adapter: lora
21
+ base_model: EleutherAI/pythia-14m
22
+ bf16: auto
23
+ chat_template: llama3
24
+ dataset_prepared_path: null
25
+ datasets:
26
+ - data_files:
27
+ - naver-news-summarization-ko_train_data.json
28
+ ds_type: json
29
+ path: /workspace/input_data/naver-news-summarization-ko_train_data.json
30
+ type:
31
+ field_input: document
32
+ field_instruction: title
33
+ field_output: summary
34
+ system_format: '{system}'
35
+ system_prompt: ''
36
+ debug: null
37
+ deepspeed: null
38
+ early_stopping_patience: null
39
+ eval_max_new_tokens: 128
40
+ eval_table_size: null
41
+ evals_per_epoch: 4
42
+ flash_attention: true
43
+ fp16: null
44
+ fsdp: null
45
+ fsdp_config: null
46
+ gradient_accumulation_steps: 4
47
+ gradient_checkpointing: true
48
+ group_by_length: false
49
+ hub_model_id: ncbateman/8f97fcaa-a2e2-4f79-8ff1-0ca9a91cd08f
50
+ hub_repo: ncbateman
51
+ hub_strategy: checkpoint
52
+ hub_token: null
53
+ learning_rate: 0.0001
54
+ load_in_4bit: false
55
+ load_in_8bit: true
56
+ local_rank: null
57
+ logging_steps: 1
58
+ lora_alpha: 32
59
+ lora_dropout: 0.05
60
+ lora_fan_in_fan_out: null
61
+ lora_model_dir: null
62
+ lora_r: 16
63
+ lora_target_linear: true
64
+ lr_scheduler: cosine
65
+ max_steps: 1000
66
+ micro_batch_size: 2
67
+ mlflow_experiment_name: /tmp/naver-news-summarization-ko_train_data.json
68
+ model_type: AutoModelForCausalLM
69
+ num_epochs: 1
70
+ optimizer: adamw_bnb_8bit
71
+ output_dir: miner_id_24
72
+ pad_to_sequence_len: true
73
+ resume_from_checkpoint: null
74
+ s2_attention: null
75
+ sample_packing: false
76
+ save_steps: 100
77
+ save_strategy: steps
78
+ sequence_len: 4096
79
+ strict: false
80
+ tf32: false
81
+ tokenizer_type: AutoTokenizer
82
+ train_on_inputs: false
83
+ val_set_size: 0.05
84
+ wandb_entity: breakfasthut
85
+ wandb_mode: online
86
+ wandb_project: tuning-miner
87
+ wandb_run: miner
88
+ wandb_runid: 8f97fcaa-a2e2-4f79-8ff1-0ca9a91cd08f
89
+ warmup_steps: 50
90
+ weight_decay: 0.0
91
+ xformers_attention: null
92
+
93
+ ```
94
+
95
+ </details><br>
96
+
97
+ # 8f97fcaa-a2e2-4f79-8ff1-0ca9a91cd08f
98
+
99
+ This model is a fine-tuned version of [EleutherAI/pythia-14m](https://huggingface.co/EleutherAI/pythia-14m) on the None dataset.
100
+ It achieves the following results on the evaluation set:
101
+ - Loss: 4.1464
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.0001
121
+ - train_batch_size: 2
122
+ - eval_batch_size: 2
123
+ - seed: 42
124
+ - distributed_type: multi-GPU
125
+ - num_devices: 4
126
+ - gradient_accumulation_steps: 4
127
+ - total_train_batch_size: 32
128
+ - total_eval_batch_size: 8
129
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
130
+ - lr_scheduler_type: cosine
131
+ - lr_scheduler_warmup_steps: 50
132
+ - training_steps: 811
133
+
134
+ ### Training results
135
+
136
+ | Training Loss | Epoch | Step | Validation Loss |
137
+ |:-------------:|:------:|:----:|:---------------:|
138
+ | 56.0062 | 0.0012 | 1 | 35.9780 |
139
+ | 4.6308 | 0.2505 | 203 | 4.4064 |
140
+ | 4.2248 | 0.5009 | 406 | 4.2050 |
141
+ | 4.0789 | 0.7514 | 609 | 4.1464 |
142
+
143
+
144
+ ### Framework versions
145
+
146
+ - PEFT 0.13.2
147
+ - Transformers 4.45.2
148
+ - Pytorch 2.4.1+cu124
149
+ - Datasets 3.0.1
150
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5d2a9b2faf38456e2844b8d6edf979d57f405060b9085ba31a1ecb6b0a35b09
3
+ size 804030