JoshMe1 commited on
Commit
78832dd
·
verified ·
1 Parent(s): 13473e0

End of training

Browse files
Files changed (2) hide show
  1. README.md +161 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,161 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: EleutherAI/pythia-160m
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: d0c65863-45bc-4f92-9929-f5e665eff8af
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: EleutherAI/pythia-160m
23
+ bf16: true
24
+ chat_template: llama3
25
+ dataset_prepared_path: null
26
+ datasets:
27
+ - data_files:
28
+ - cb60d6a9b4ff3d3a_train_data.json
29
+ ds_type: json
30
+ field: query
31
+ path: /workspace/input_data/cb60d6a9b4ff3d3a_train_data.json
32
+ type: completion
33
+ debug: null
34
+ deepspeed: null
35
+ device_map: auto
36
+ early_stopping_patience: 3
37
+ ema_decay: 0.9992
38
+ eval_max_new_tokens: 128
39
+ eval_steps: 100
40
+ eval_table_size: null
41
+ evals_per_epoch: null
42
+ flash_attention: true
43
+ fp16: false
44
+ fsdp: null
45
+ fsdp_config: null
46
+ gradient_accumulation_steps: 4
47
+ gradient_checkpointing: true
48
+ greater_is_better: false
49
+ group_by_length: false
50
+ hub_model_id: JoshMe1/d0c65863-45bc-4f92-9929-f5e665eff8af
51
+ hub_repo: null
52
+ hub_strategy: checkpoint
53
+ hub_token: null
54
+ learning_rate: 0.0001
55
+ load_best_model_at_end: true
56
+ load_in_4bit: false
57
+ load_in_8bit: true
58
+ local_rank: null
59
+ logging_steps: 10
60
+ lora_alpha: 256
61
+ lora_dropout: 0.1
62
+ lora_fan_in_fan_out: null
63
+ lora_model_dir: null
64
+ lora_r: 64
65
+ lora_target_linear: true
66
+ lr_scheduler: reduce_lr_on_plateau
67
+ lr_scheduler_factor: 0.5
68
+ lr_scheduler_patience: 2
69
+ max_memory:
70
+ 0: 130GB
71
+ max_steps: 500
72
+ metric_for_best_model: eval_loss
73
+ micro_batch_size: 4
74
+ mlflow_experiment_name: /tmp/cb60d6a9b4ff3d3a_train_data.json
75
+ model_type: AutoModelForCausalLM
76
+ num_epochs: 1
77
+ optimizer: adamw_hf
78
+ output_dir: miner_id_24
79
+ pad_to_sequence_len: true
80
+ resume_from_checkpoint: null
81
+ s2_attention: null
82
+ sample_packing: false
83
+ save_steps: 100
84
+ saves_per_epoch: null
85
+ sequence_len: 2048
86
+ special_tokens:
87
+ pad_token: <|endoftext|>
88
+ strict: false
89
+ tf32: false
90
+ tokenizer_type: AutoTokenizer
91
+ train_on_inputs: false
92
+ trust_remote_code: true
93
+ use_ema: true
94
+ val_set_size: 0.05
95
+ wandb_entity: null
96
+ wandb_mode: online
97
+ wandb_name: 72033cbb-9607-4a6e-a252-62a89a2f99d4
98
+ wandb_project: Gradients-On-Demand
99
+ wandb_run: your_name
100
+ wandb_runid: 72033cbb-9607-4a6e-a252-62a89a2f99d4
101
+ warmup_steps: 200
102
+ weight_decay: 0.01
103
+ xformers_attention: null
104
+
105
+ ```
106
+
107
+ </details><br>
108
+
109
+ # d0c65863-45bc-4f92-9929-f5e665eff8af
110
+
111
+ This model is a fine-tuned version of [EleutherAI/pythia-160m](https://huggingface.co/EleutherAI/pythia-160m) on the None dataset.
112
+ It achieves the following results on the evaluation set:
113
+ - Loss: 4.9802
114
+
115
+ ## Model description
116
+
117
+ More information needed
118
+
119
+ ## Intended uses & limitations
120
+
121
+ More information needed
122
+
123
+ ## Training and evaluation data
124
+
125
+ More information needed
126
+
127
+ ## Training procedure
128
+
129
+ ### Training hyperparameters
130
+
131
+ The following hyperparameters were used during training:
132
+ - learning_rate: 0.0001
133
+ - train_batch_size: 4
134
+ - eval_batch_size: 4
135
+ - seed: 42
136
+ - gradient_accumulation_steps: 4
137
+ - total_train_batch_size: 16
138
+ - optimizer: Use OptimizerNames.ADAMW_HF with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
139
+ - lr_scheduler_type: reduce_lr_on_plateau
140
+ - lr_scheduler_warmup_steps: 200
141
+ - training_steps: 500
142
+
143
+ ### Training results
144
+
145
+ | Training Loss | Epoch | Step | Validation Loss |
146
+ |:-------------:|:------:|:----:|:---------------:|
147
+ | No log | 0.0001 | 1 | 4.8533 |
148
+ | 21.894 | 0.0085 | 100 | 5.8594 |
149
+ | 22.1192 | 0.0169 | 200 | 5.7406 |
150
+ | 20.4868 | 0.0254 | 300 | 5.1533 |
151
+ | 20.419 | 0.0339 | 400 | 5.1123 |
152
+ | 20.3705 | 0.0423 | 500 | 4.9802 |
153
+
154
+
155
+ ### Framework versions
156
+
157
+ - PEFT 0.13.2
158
+ - Transformers 4.46.0
159
+ - Pytorch 2.5.0+cu124
160
+ - Datasets 3.0.1
161
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63501361655d391fa136c1199b5fd3082b61fa7a457ccb510662fb7df3bc3d6a
3
+ size 37783950