Lin2es commited on
Commit
96174d3
·
verified ·
1 Parent(s): 6df9933

End of training

Browse files
Files changed (2) hide show
  1. README.md +159 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,159 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: EleutherAI/pythia-14m
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: 4a169dfd-a170-48ce-a7fa-aa4b0a42df87
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
+ - 94a3f1dc5561b4c3_train_data.json
28
+ ds_type: json
29
+ format: custom
30
+ path: /workspace/input_data/94a3f1dc5561b4c3_train_data.json
31
+ type:
32
+ field_input: prompt
33
+ field_instruction: instruction
34
+ field_output: response
35
+ format: '{instruction} {input}'
36
+ no_input_format: '{instruction}'
37
+ system_format: '{system}'
38
+ system_prompt: ''
39
+ debug: null
40
+ deepspeed: null
41
+ do_eval: true
42
+ early_stopping_patience: 3
43
+ eval_batch_size: 4
44
+ eval_max_new_tokens: 128
45
+ eval_steps: 250
46
+ evals_per_epoch: null
47
+ flash_attention: true
48
+ fp16: false
49
+ fsdp: null
50
+ fsdp_config: null
51
+ gradient_accumulation_steps: 8
52
+ gradient_checkpointing: true
53
+ group_by_length: true
54
+ hub_model_id: Lin2es/4a169dfd-a170-48ce-a7fa-aa4b0a42df87
55
+ hub_repo: null
56
+ hub_strategy: checkpoint
57
+ hub_token: null
58
+ learning_rate: 0.0002
59
+ load_in_4bit: false
60
+ load_in_8bit: false
61
+ local_rank: null
62
+ logging_steps: 50
63
+ lora_alpha: 32
64
+ lora_dropout: 0.1
65
+ lora_fan_in_fan_out: null
66
+ lora_model_dir: null
67
+ lora_r: 16
68
+ lora_target_linear: true
69
+ lr_scheduler: cosine
70
+ max_grad_norm: 1.0
71
+ max_steps: 500
72
+ micro_batch_size: 4
73
+ mlflow_experiment_name: /tmp/94a3f1dc5561b4c3_train_data.json
74
+ model_type: AutoModelForCausalLM
75
+ num_epochs: 1
76
+ optimizer: adamw_torch_fused
77
+ output_dir: miner_id_24
78
+ pad_to_sequence_len: true
79
+ resume_from_checkpoint: null
80
+ s2_attention: null
81
+ sample_packing: false
82
+ save_steps: 500
83
+ saves_per_epoch: null
84
+ seed: 1001
85
+ sequence_len: 1024
86
+ special_tokens:
87
+ pad_token: <|endoftext|>
88
+ strict: false
89
+ tf32: true
90
+ tokenizer_type: AutoTokenizer
91
+ train_on_inputs: false
92
+ trust_remote_code: true
93
+ val_set_size: 0.05
94
+ wandb_entity: null
95
+ wandb_mode: online
96
+ wandb_name: 3b9b881b-877f-40bb-8dd0-f5683c1fae9d
97
+ wandb_project: text01
98
+ wandb_run: your_name
99
+ wandb_runid: 3b9b881b-877f-40bb-8dd0-f5683c1fae9d
100
+ warmup_steps: 100
101
+ weight_decay: 0.0
102
+ xformers_attention: null
103
+
104
+ ```
105
+
106
+ </details><br>
107
+
108
+ # 4a169dfd-a170-48ce-a7fa-aa4b0a42df87
109
+
110
+ This model is a fine-tuned version of [EleutherAI/pythia-14m](https://huggingface.co/EleutherAI/pythia-14m) on the None dataset.
111
+ It achieves the following results on the evaluation set:
112
+ - Loss: 3.9559
113
+
114
+ ## Model description
115
+
116
+ More information needed
117
+
118
+ ## Intended uses & limitations
119
+
120
+ More information needed
121
+
122
+ ## Training and evaluation data
123
+
124
+ More information needed
125
+
126
+ ## Training procedure
127
+
128
+ ### Training hyperparameters
129
+
130
+ The following hyperparameters were used during training:
131
+ - learning_rate: 0.0002
132
+ - train_batch_size: 4
133
+ - eval_batch_size: 4
134
+ - seed: 1001
135
+ - distributed_type: multi-GPU
136
+ - num_devices: 4
137
+ - gradient_accumulation_steps: 8
138
+ - total_train_batch_size: 128
139
+ - total_eval_batch_size: 16
140
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
141
+ - lr_scheduler_type: cosine
142
+ - lr_scheduler_warmup_steps: 100
143
+ - training_steps: 369
144
+
145
+ ### Training results
146
+
147
+ | Training Loss | Epoch | Step | Validation Loss |
148
+ |:-------------:|:------:|:----:|:---------------:|
149
+ | No log | 0.0027 | 1 | 5.2547 |
150
+ | 32.7118 | 0.6782 | 250 | 3.9559 |
151
+
152
+
153
+ ### Framework versions
154
+
155
+ - PEFT 0.13.2
156
+ - Transformers 4.46.0
157
+ - Pytorch 2.5.0+cu124
158
+ - Datasets 3.0.1
159
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:438f0009ef1934c0e8d733220fac669dc47eb72416f6d567b9ec8cd2a94c9370
3
+ size 804030