jtatman commited on
Commit
d4f00fa
1 Parent(s): 6b5f201

End of training

Browse files
README.md ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Locutusque/TinyMistral-248M
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: tinymistral-248-DPO
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
+ # tinymistral-248-DPO
17
+
18
+ This model is a fine-tuned version of [Locutusque/TinyMistral-248M](https://huggingface.co/Locutusque/TinyMistral-248M) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.3205
21
+ - Accuracy: 0.0
22
+ - Rewards/chosen: 0.7722
23
+ - Rewards/rejected: -0.2727
24
+ - Rewards/accuracies: 1.0
25
+ - Rewards/margins: 1.0449
26
+ - Logps/rejected: -286.5494
27
+ - Logps/chosen: -398.5646
28
+ - Logits/rejected: -2.3562
29
+ - Logits/chosen: -1.8620
30
+
31
+ ## Model description
32
+
33
+ More information needed
34
+
35
+ ## Intended uses & limitations
36
+
37
+ More information needed
38
+
39
+ ## Training and evaluation data
40
+
41
+ More information needed
42
+
43
+ ## Training procedure
44
+
45
+ ### Training hyperparameters
46
+
47
+ The following hyperparameters were used during training:
48
+ - learning_rate: 0.0002
49
+ - train_batch_size: 12
50
+ - eval_batch_size: 1
51
+ - seed: 42
52
+ - gradient_accumulation_steps: 12
53
+ - total_train_batch_size: 144
54
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
+ - lr_scheduler_type: constant
56
+ - lr_scheduler_warmup_ratio: 0.05
57
+ - num_epochs: 6
58
+ - mixed_precision_training: Native AMP
59
+
60
+ ### Training results
61
+
62
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
63
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
64
+ | 0.5815 | 0.48 | 10 | 0.3205 | 0.0 | 0.7722 | -0.2727 | 1.0 | 1.0449 | -286.5494 | -398.5646 | -2.3562 | -1.8620 |
65
+ | 0.3287 | 0.95 | 20 | 0.0970 | 0.0 | 1.0191 | -1.8694 | 1.0 | 2.8886 | -302.5168 | -396.0956 | -2.0547 | -1.5790 |
66
+ | 0.2126 | 1.43 | 30 | 0.0414 | 0.0 | 0.3685 | -4.5314 | 1.0 | 4.8999 | -329.1370 | -402.6024 | -1.8100 | -1.4099 |
67
+ | 0.1844 | 1.9 | 40 | 0.0260 | 0.0 | 0.9879 | -4.8275 | 1.0 | 5.8153 | -332.0973 | -396.4084 | -1.8704 | -1.4976 |
68
+ | 0.1546 | 2.38 | 50 | 0.0190 | 0.0 | 1.1813 | -5.2560 | 1.0 | 6.4373 | -336.3821 | -394.4740 | -1.9098 | -1.5582 |
69
+ | 0.1532 | 2.86 | 60 | 0.0140 | 0.0 | 1.0583 | -6.0198 | 1.0 | 7.0780 | -344.0201 | -395.7045 | -1.8920 | -1.5654 |
70
+ | 0.1402 | 3.33 | 70 | 0.0112 | 0.0 | 1.0134 | -6.5382 | 1.0 | 7.5517 | -349.2049 | -396.1526 | -1.8823 | -1.5706 |
71
+ | 0.1544 | 3.81 | 80 | 0.0089 | 0.0 | 0.8836 | -7.1726 | 1.0 | 8.0562 | -355.5490 | -397.4513 | -1.8518 | -1.5535 |
72
+ | 0.1357 | 4.29 | 90 | 0.0072 | 0.0 | 0.7532 | -7.7663 | 1.0 | 8.5195 | -361.4852 | -398.7546 | -1.8193 | -1.5345 |
73
+ | 0.1418 | 4.76 | 100 | 0.0061 | 0.0 | 0.6041 | -8.3133 | 1.0 | 8.9174 | -366.9556 | -400.2459 | -1.7889 | -1.5150 |
74
+ | 0.1482 | 5.24 | 110 | 0.0051 | 0.0 | 0.4867 | -8.7961 | 1.0 | 9.2828 | -371.7837 | -401.4203 | -1.7611 | -1.4971 |
75
+ | 0.141 | 5.71 | 120 | 0.0045 | 0.0 | 0.4212 | -9.1494 | 1.0 | 9.5706 | -375.3166 | -402.0751 | -1.7409 | -1.4842 |
76
+
77
+
78
+ ### Framework versions
79
+
80
+ - Transformers 4.35.2
81
+ - Pytorch 2.1.0+cu121
82
+ - Datasets 2.15.0
83
+ - Tokenizers 0.15.0
adapter_config.json ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "Locutusque/TinyMistral-248M",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "loftq_config": {},
12
+ "lora_alpha": 16,
13
+ "lora_dropout": 0.05,
14
+ "megatron_config": null,
15
+ "megatron_core": "megatron.core",
16
+ "modules_to_save": null,
17
+ "peft_type": "LORA",
18
+ "r": 16,
19
+ "rank_pattern": {},
20
+ "revision": null,
21
+ "target_modules": [
22
+ "model.layers.2.self_attn.k_proj",
23
+ "model.layers.8.mlp.down_proj",
24
+ "model.layers.8.mlp.up_proj",
25
+ "model.layers.7.self_attn.o_proj",
26
+ "model.layers.6.self_attn.k_proj",
27
+ "model.layers.6.self_attn.o_proj",
28
+ "model.layers.4.mlp.gate_proj",
29
+ "model.layers.4.self_attn.q_proj",
30
+ "model.layers.6.mlp.up_proj",
31
+ "model.layers.6.mlp.down_proj",
32
+ "model.layers.2.mlp.gate_proj",
33
+ "model.layers.5.self_attn.v_proj",
34
+ "model.layers.7.mlp.down_proj",
35
+ "model.layers.7.mlp.up_proj",
36
+ "model.layers.1.self_attn.o_proj",
37
+ "lm_head",
38
+ "model.layers.5.mlp.up_proj",
39
+ "model.layers.0.mlp.gate_proj",
40
+ "model.layers.9.self_attn.k_proj",
41
+ "model.layers.2.self_attn.o_proj",
42
+ "model.layers.4.mlp.up_proj",
43
+ "model.layers.1.self_attn.k_proj",
44
+ "model.layers.3.self_attn.v_proj",
45
+ "model.layers.10.mlp.down_proj",
46
+ "model.layers.10.mlp.gate_proj",
47
+ "model.layers.6.self_attn.q_proj",
48
+ "model.layers.3.self_attn.k_proj",
49
+ "model.layers.9.mlp.up_proj",
50
+ "model.layers.7.self_attn.k_proj",
51
+ "model.layers.3.self_attn.q_proj",
52
+ "model.layers.8.self_attn.o_proj",
53
+ "model.layers.10.mlp.up_proj",
54
+ "model.layers.7.self_attn.q_proj",
55
+ "model.layers.0.mlp.down_proj",
56
+ "model.layers.7.mlp.gate_proj",
57
+ "model.layers.3.mlp.down_proj",
58
+ "model.layers.10.self_attn.o_proj",
59
+ "model.layers.4.self_attn.o_proj",
60
+ "model.layers.3.self_attn.o_proj",
61
+ "model.layers.5.self_attn.k_proj",
62
+ "model.layers.1.self_attn.q_proj",
63
+ "model.layers.5.self_attn.o_proj",
64
+ "model.layers.11.mlp.down_proj",
65
+ "model.layers.9.self_attn.v_proj",
66
+ "model.layers.1.self_attn.v_proj",
67
+ "model.layers.2.self_attn.v_proj",
68
+ "model.layers.4.mlp.down_proj",
69
+ "model.layers.10.self_attn.q_proj",
70
+ "model.layers.0.self_attn.q_proj",
71
+ "model.layers.3.mlp.gate_proj",
72
+ "model.layers.9.self_attn.o_proj",
73
+ "model.layers.8.self_attn.q_proj",
74
+ "model.layers.10.self_attn.v_proj",
75
+ "model.layers.3.mlp.up_proj",
76
+ "model.layers.8.self_attn.v_proj",
77
+ "model.layers.8.self_attn.k_proj",
78
+ "model.layers.9.mlp.down_proj",
79
+ "model.layers.11.mlp.up_proj",
80
+ "model.layers.4.self_attn.k_proj",
81
+ "model.layers.11.self_attn.v_proj",
82
+ "model.layers.10.self_attn.k_proj",
83
+ "model.layers.0.mlp.up_proj",
84
+ "model.layers.1.mlp.down_proj",
85
+ "model.layers.11.self_attn.q_proj",
86
+ "model.layers.1.mlp.gate_proj",
87
+ "model.layers.5.self_attn.q_proj",
88
+ "model.layers.7.self_attn.v_proj",
89
+ "model.layers.5.mlp.down_proj",
90
+ "model.layers.9.mlp.gate_proj",
91
+ "model.layers.11.self_attn.k_proj",
92
+ "model.layers.9.self_attn.q_proj",
93
+ "model.layers.2.mlp.up_proj",
94
+ "model.layers.4.self_attn.v_proj",
95
+ "model.layers.11.self_attn.o_proj",
96
+ "model.layers.1.mlp.up_proj",
97
+ "model.layers.0.self_attn.k_proj",
98
+ "model.layers.5.mlp.gate_proj",
99
+ "model.layers.6.mlp.gate_proj",
100
+ "model.layers.8.mlp.gate_proj",
101
+ "model.layers.0.self_attn.v_proj",
102
+ "model.layers.6.self_attn.v_proj",
103
+ "model.layers.2.self_attn.q_proj",
104
+ "model.layers.2.mlp.down_proj",
105
+ "model.layers.0.self_attn.o_proj",
106
+ "model.layers.11.mlp.gate_proj"
107
+ ],
108
+ "task_type": "CAUSAL_LM",
109
+ "use_rslora": false
110
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebe9a629dd2599614e3a4d28887a66a753a92b950dfd372934bd458239fca533
3
+ size 150137592
special_tokens_map.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|ASSISTANT|>",
4
+ "<|USER|>"
5
+ ],
6
+ "bos_token": {
7
+ "content": "<|bos|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false
12
+ },
13
+ "eos_token": {
14
+ "content": "<|endoftext|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false
19
+ },
20
+ "pad_token": {
21
+ "content": "[PAD]",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ },
27
+ "unk_token": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false
33
+ }
34
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<unk>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<s>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "32000": {
28
+ "content": "<|bos|>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "32001": {
36
+ "content": "<|endoftext|>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "32002": {
44
+ "content": "[PAD]",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "32003": {
52
+ "content": "<|ASSISTANT|>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "32004": {
60
+ "content": "<|USER|>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ }
67
+ },
68
+ "additional_special_tokens": [
69
+ "<|ASSISTANT|>",
70
+ "<|USER|>"
71
+ ],
72
+ "bos_token": "<|bos|>",
73
+ "clean_up_tokenization_spaces": false,
74
+ "eos_token": "<|endoftext|>",
75
+ "legacy": true,
76
+ "max_length": 1536,
77
+ "model_max_length": 1000000000000000019884624838656,
78
+ "pad_token": "[PAD]",
79
+ "sp_model_kwargs": {},
80
+ "spaces_between_special_tokens": false,
81
+ "stride": 0,
82
+ "tokenizer_class": "LlamaTokenizer",
83
+ "truncation_side": "right",
84
+ "truncation_strategy": "longest_first",
85
+ "unk_token": "<unk>",
86
+ "use_default_system_prompt": true
87
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1490c0eac1bb220a0df75e33af316b4aee8b487edfafd47a0ccf5bec2c878634
3
+ size 4600