minpeter commited on
Commit
ec5f789
·
verified ·
1 Parent(s): 44627b2

End of training

Browse files
Files changed (2) hide show
  1. README.md +122 -0
  2. generation_config.json +8 -0
README.md ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ base_model: minpeter/pretrained-tiny-ko
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ datasets:
8
+ - lemon-mint/Korean-FineTome-100k
9
+ model-index:
10
+ - name: ko-tiny-exp
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ [<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)
18
+ <details><summary>See axolotl config</summary>
19
+
20
+ axolotl version: `0.10.0.dev0`
21
+ ```yaml
22
+ base_model: minpeter/pretrained-tiny-ko
23
+
24
+ chat_template: tokenizer_default_fallback_chatml
25
+ datasets:
26
+ - path: lemon-mint/Korean-FineTome-100k
27
+ type: chat_template
28
+ split: train[:20%]
29
+ field_messages: messages
30
+ message_property_mappings:
31
+ role: role
32
+ content: content
33
+ dataset_prepared_path: last_run_prepared
34
+ val_set_size: 0.05
35
+
36
+ hub_model_id: minpeter/ko-tiny-exp
37
+ output_dir: ./ouputs/ko-tiny-exp
38
+ wandb_project: "axolotl"
39
+ wandb_entity: "kasfiekfs-e"
40
+
41
+ sequence_len: 2048
42
+ sample_packing: true
43
+ pad_to_sequence_len: true
44
+
45
+ gradient_accumulation_steps: 4
46
+ micro_batch_size: 16
47
+
48
+ num_epochs: 2
49
+ optimizer: paged_adamw_8bit
50
+ lr_scheduler: cosine
51
+ learning_rate: 2e-5
52
+
53
+ bf16: auto
54
+ tf32: false
55
+
56
+ gradient_checkpointing: true
57
+ gradient_checkpointing_kwargs:
58
+ use_reentrant: false
59
+ resume_from_checkpoint:
60
+ logging_steps: 1
61
+ flash_attention: true
62
+
63
+ warmup_steps: 100
64
+ evals_per_epoch: 2
65
+ saves_per_epoch: 1
66
+ weight_decay: 0.0
67
+
68
+ ```
69
+
70
+ </details><br>
71
+
72
+ # ko-tiny-exp
73
+
74
+ This model is a fine-tuned version of [minpeter/pretrained-tiny-ko](https://huggingface.co/minpeter/pretrained-tiny-ko) on the lemon-mint/Korean-FineTome-100k dataset.
75
+ It achieves the following results on the evaluation set:
76
+ - Loss: 2.9471
77
+
78
+ ## Model description
79
+
80
+ More information needed
81
+
82
+ ## Intended uses & limitations
83
+
84
+ More information needed
85
+
86
+ ## Training and evaluation data
87
+
88
+ More information needed
89
+
90
+ ## Training procedure
91
+
92
+ ### Training hyperparameters
93
+
94
+ The following hyperparameters were used during training:
95
+ - learning_rate: 2e-05
96
+ - train_batch_size: 16
97
+ - eval_batch_size: 16
98
+ - seed: 42
99
+ - gradient_accumulation_steps: 4
100
+ - total_train_batch_size: 64
101
+ - optimizer: Use OptimizerNames.PAGED_ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
102
+ - lr_scheduler_type: cosine
103
+ - lr_scheduler_warmup_steps: 100
104
+ - training_steps: 176
105
+
106
+ ### Training results
107
+
108
+ | Training Loss | Epoch | Step | Validation Loss |
109
+ |:-------------:|:------:|:----:|:---------------:|
110
+ | 3.0538 | 0.0113 | 1 | 3.0136 |
111
+ | 2.9454 | 0.4972 | 44 | 3.0099 |
112
+ | 2.9657 | 0.9944 | 88 | 2.9874 |
113
+ | 3.0426 | 1.4859 | 132 | 2.9506 |
114
+ | 2.9403 | 1.9831 | 176 | 2.9471 |
115
+
116
+
117
+ ### Framework versions
118
+
119
+ - Transformers 4.51.3
120
+ - Pytorch 2.6.0+cu124
121
+ - Datasets 3.5.1
122
+ - Tokenizers 0.21.1
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 128256,
4
+ "do_sample": true,
5
+ "eos_token_id": 128257,
6
+ "transformers_version": "4.51.3",
7
+ "use_cache": false
8
+ }