SystemAdmin123 commited on
Commit
098e828
·
verified ·
1 Parent(s): 6cd43c6

End of training

Browse files
Files changed (2) hide show
  1. README.md +126 -0
  2. generation_config.json +7 -0
README.md ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: unsloth/SmolLM-360M
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ datasets:
9
+ - argilla/databricks-dolly-15k-curated-en
10
+ model-index:
11
+ - name: SmolLM-360M
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ [<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)
19
+ <details><summary>See axolotl config</summary>
20
+
21
+ axolotl version: `0.6.0`
22
+ ```yaml
23
+ base_model: unsloth/SmolLM-360M
24
+ batch_size: 92
25
+ bf16: true
26
+ chat_template: tokenizer_default_fallback_alpaca
27
+ datasets:
28
+ - format: custom
29
+ path: argilla/databricks-dolly-15k-curated-en
30
+ type:
31
+ field_input: original-instruction
32
+ field_instruction: original-instruction
33
+ field_output: original-response
34
+ format: '{instruction} {input}'
35
+ no_input_format: '{instruction}'
36
+ system_format: '{system}'
37
+ system_prompt: ''
38
+ device_map: auto
39
+ eval_sample_packing: false
40
+ eval_steps: 200
41
+ flash_attention: true
42
+ gradient_checkpointing: true
43
+ group_by_length: true
44
+ hub_model_id: SystemAdmin123/SmolLM-360M
45
+ hub_strategy: checkpoint
46
+ learning_rate: 0.0002
47
+ logging_steps: 10
48
+ lr_scheduler: cosine
49
+ max_steps: 10000
50
+ micro_batch_size: 23
51
+ model_type: AutoModelForCausalLM
52
+ num_epochs: 100
53
+ optimizer: adamw_bnb_8bit
54
+ output_dir: /root/.sn56/axolotl/tmp/SmolLM-360M
55
+ pad_to_sequence_len: true
56
+ resize_token_embeddings_to_32x: false
57
+ sample_packing: true
58
+ save_steps: 200
59
+ save_total_limit: 1
60
+ sequence_len: 2048
61
+ tokenizer_type: GPT2TokenizerFast
62
+ torch_dtype: bf16
63
+ trust_remote_code: true
64
+ val_set_size: 0.1
65
+ wandb_entity: ''
66
+ wandb_mode: online
67
+ wandb_name: unsloth/SmolLM-360M-argilla/databricks-dolly-15k-curated-en
68
+ wandb_project: Gradients-On-Demand
69
+ wandb_run: your_name
70
+ wandb_runid: default
71
+ warmup_ratio: 0.05
72
+
73
+ ```
74
+
75
+ </details><br>
76
+
77
+ # SmolLM-360M
78
+
79
+ This model is a fine-tuned version of [unsloth/SmolLM-360M](https://huggingface.co/unsloth/SmolLM-360M) on the argilla/databricks-dolly-15k-curated-en dataset.
80
+ It achieves the following results on the evaluation set:
81
+ - Loss: 2.0617
82
+
83
+ ## Model description
84
+
85
+ More information needed
86
+
87
+ ## Intended uses & limitations
88
+
89
+ More information needed
90
+
91
+ ## Training and evaluation data
92
+
93
+ More information needed
94
+
95
+ ## Training procedure
96
+
97
+ ### Training hyperparameters
98
+
99
+ The following hyperparameters were used during training:
100
+ - learning_rate: 0.0002
101
+ - train_batch_size: 23
102
+ - eval_batch_size: 23
103
+ - seed: 42
104
+ - distributed_type: multi-GPU
105
+ - num_devices: 4
106
+ - total_train_batch_size: 92
107
+ - total_eval_batch_size: 92
108
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
109
+ - lr_scheduler_type: cosine
110
+ - lr_scheduler_warmup_steps: 10
111
+ - training_steps: 200
112
+
113
+ ### Training results
114
+
115
+ | Training Loss | Epoch | Step | Validation Loss |
116
+ |:-------------:|:-----:|:----:|:---------------:|
117
+ | No log | 0.125 | 1 | 2.5584 |
118
+ | 2.011 | 25.0 | 200 | 2.0617 |
119
+
120
+
121
+ ### Framework versions
122
+
123
+ - Transformers 4.48.1
124
+ - Pytorch 2.5.1+cu124
125
+ - Datasets 3.2.0
126
+ - Tokenizers 0.21.0
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "do_sample": true,
5
+ "eos_token_id": 0,
6
+ "transformers_version": "4.48.1"
7
+ }