BleachNick commited on
Commit
fa9564c
·
verified ·
1 Parent(s): 0b83cf8

Model save

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen2-VL-2B-Instruct
3
+ library_name: transformers
4
+ model_name: Qwen2-VL-2B-Instruct-SFT_1600_epo20
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for Qwen2-VL-2B-Instruct-SFT_1600_epo20
13
+
14
+ This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="BleachNick/Qwen2-VL-2B-Instruct-SFT_1600_epo20", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/haozhezhao/huggingface/runs/nrhznie5)
31
+
32
+
33
+ This model was trained with SFT.
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.15.0.dev0
38
+ - Transformers: 4.49.0.dev0
39
+ - Pytorch: 2.5.1+cu124
40
+ - Datasets: 3.2.0
41
+ - Tokenizers: 0.21.0
42
+
43
+ ## Citations
44
+
45
+
46
+
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
added_tokens.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "<|box_end|>": 151649,
3
+ "<|box_start|>": 151648,
4
+ "<|endoftext|>": 151643,
5
+ "<|im_end|>": 151645,
6
+ "<|im_start|>": 151644,
7
+ "<|image_pad|>": 151655,
8
+ "<|object_ref_end|>": 151647,
9
+ "<|object_ref_start|>": 151646,
10
+ "<|quad_end|>": 151651,
11
+ "<|quad_start|>": 151650,
12
+ "<|video_pad|>": 151656,
13
+ "<|vision_end|>": 151653,
14
+ "<|vision_pad|>": 151654,
15
+ "<|vision_start|>": 151652
16
+ }
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 8.00754383150121e+16,
4
+ "train_loss": 0.29854341844717663,
5
+ "train_runtime": 5255.3389,
6
+ "train_samples": 1600,
7
+ "train_samples_per_second": 3.045,
8
+ "train_steps_per_second": 0.057
9
+ }
config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Qwen/Qwen2-VL-2B-Instruct",
3
+ "architectures": [
4
+ "Qwen2VLForConditionalGeneration"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "eos_token_id": 151645,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 1536,
11
+ "image_token_id": 151655,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 8960,
14
+ "max_position_embeddings": 32768,
15
+ "max_window_layers": 28,
16
+ "model_type": "qwen2_vl",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 28,
19
+ "num_key_value_heads": 2,
20
+ "rms_norm_eps": 1e-06,
21
+ "rope_scaling": {
22
+ "mrope_section": [
23
+ 16,
24
+ 24,
25
+ 24
26
+ ],
27
+ "rope_type": "default",
28
+ "type": "default"
29
+ },
30
+ "rope_theta": 1000000.0,
31
+ "sliding_window": 32768,
32
+ "tie_word_embeddings": true,
33
+ "torch_dtype": "bfloat16",
34
+ "transformers_version": "4.49.0.dev0",
35
+ "use_cache": false,
36
+ "use_sliding_window": false,
37
+ "video_token_id": 151656,
38
+ "vision_config": {
39
+ "hidden_size": 1536,
40
+ "in_chans": 3,
41
+ "model_type": "qwen2_vl",
42
+ "spatial_patch_size": 14,
43
+ "torch_dtype": "bfloat16"
44
+ },
45
+ "vision_end_token_id": 151653,
46
+ "vision_start_token_id": 151652,
47
+ "vision_token_id": 151654,
48
+ "vocab_size": 151936
49
+ }
generation_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_implementation": null,
3
+ "bos_token_id": 151643,
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 151645,
7
+ 151643
8
+ ],
9
+ "pad_token_id": 151643,
10
+ "temperature": 0.01,
11
+ "top_k": 1,
12
+ "top_p": 0.001,
13
+ "transformers_version": "4.49.0.dev0",
14
+ "use_cache": false
15
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b049be79a68acd5cf558694f7916b319ac15dd51556f57684477ed86e497ad76
3
+ size 4884798456
special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|object_ref_start|>",
6
+ "<|object_ref_end|>",
7
+ "<|box_start|>",
8
+ "<|box_end|>",
9
+ "<|quad_start|>",
10
+ "<|quad_end|>",
11
+ "<|vision_start|>",
12
+ "<|vision_end|>",
13
+ "<|vision_pad|>",
14
+ "<|image_pad|>",
15
+ "<|video_pad|>"
16
+ ],
17
+ "eos_token": {
18
+ "content": "<|im_end|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88a3a6fcb80132f76da8aa40cdc3fccd7e5d8468ef15421f5b0c2715e85217d2
3
+ size 11420538
tokenizer_config.json ADDED
@@ -0,0 +1,144 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "151643": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "151644": {
13
+ "content": "<|im_start|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "151645": {
21
+ "content": "<|im_end|>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "151646": {
29
+ "content": "<|object_ref_start|>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "151647": {
37
+ "content": "<|object_ref_end|>",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "151648": {
45
+ "content": "<|box_start|>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "151649": {
53
+ "content": "<|box_end|>",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "151650": {
61
+ "content": "<|quad_start|>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "151651": {
69
+ "content": "<|quad_end|>",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "151652": {
77
+ "content": "<|vision_start|>",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "151653": {
85
+ "content": "<|vision_end|>",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "151654": {
93
+ "content": "<|vision_pad|>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "151655": {
101
+ "content": "<|image_pad|>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ },
108
+ "151656": {
109
+ "content": "<|video_pad|>",
110
+ "lstrip": false,
111
+ "normalized": false,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": true
115
+ }
116
+ },
117
+ "additional_special_tokens": [
118
+ "<|im_start|>",
119
+ "<|im_end|>",
120
+ "<|object_ref_start|>",
121
+ "<|object_ref_end|>",
122
+ "<|box_start|>",
123
+ "<|box_end|>",
124
+ "<|quad_start|>",
125
+ "<|quad_end|>",
126
+ "<|vision_start|>",
127
+ "<|vision_end|>",
128
+ "<|vision_pad|>",
129
+ "<|image_pad|>",
130
+ "<|video_pad|>"
131
+ ],
132
+ "bos_token": null,
133
+ "chat_template": "{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n{% endif %}<|im_start|>{{ message['role'] }}\n{% if message['content'] is string %}{{ message['content'] }}<|im_end|>\n{% else %}{% for content in message['content'] %}{% if content['type'] == 'image' or 'image' in content or 'image_url' in content %}{% set image_count.value = image_count.value + 1 %}{% if add_vision_id %}Picture {{ image_count.value }}: {% endif %}<|vision_start|><|image_pad|><|vision_end|>{% elif content['type'] == 'video' or 'video' in content %}{% set video_count.value = video_count.value + 1 %}{% if add_vision_id %}Video {{ video_count.value }}: {% endif %}<|vision_start|><|video_pad|><|vision_end|>{% elif 'text' in content %}{{ content['text'] }}{% endif %}{% endfor %}<|im_end|>\n{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant\n{% endif %}",
134
+ "clean_up_tokenization_spaces": false,
135
+ "eos_token": "<|im_end|>",
136
+ "errors": "replace",
137
+ "extra_special_tokens": {},
138
+ "model_max_length": 32768,
139
+ "pad_token": "<|endoftext|>",
140
+ "padding_side": "left",
141
+ "split_special_tokens": false,
142
+ "tokenizer_class": "Qwen2Tokenizer",
143
+ "unk_token": null
144
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 8.00754383150121e+16,
4
+ "train_loss": 0.29854341844717663,
5
+ "train_runtime": 5255.3389,
6
+ "train_samples": 1600,
7
+ "train_samples_per_second": 3.045,
8
+ "train_steps_per_second": 0.057
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,462 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 300,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.16666666666666666,
13
+ "grad_norm": 39.13534927368164,
14
+ "learning_rate": 1.6666666666666667e-06,
15
+ "loss": 2.3532,
16
+ "step": 5
17
+ },
18
+ {
19
+ "epoch": 0.3333333333333333,
20
+ "grad_norm": 17.555816650390625,
21
+ "learning_rate": 3.3333333333333333e-06,
22
+ "loss": 1.9868,
23
+ "step": 10
24
+ },
25
+ {
26
+ "epoch": 0.5,
27
+ "grad_norm": 8.326282501220703,
28
+ "learning_rate": 5e-06,
29
+ "loss": 1.259,
30
+ "step": 15
31
+ },
32
+ {
33
+ "epoch": 0.6666666666666666,
34
+ "grad_norm": 5.493110179901123,
35
+ "learning_rate": 6.666666666666667e-06,
36
+ "loss": 0.8018,
37
+ "step": 20
38
+ },
39
+ {
40
+ "epoch": 0.8333333333333334,
41
+ "grad_norm": 8.113493919372559,
42
+ "learning_rate": 8.333333333333334e-06,
43
+ "loss": 0.6479,
44
+ "step": 25
45
+ },
46
+ {
47
+ "epoch": 1.0,
48
+ "grad_norm": 3.912256956100464,
49
+ "learning_rate": 1e-05,
50
+ "loss": 0.5196,
51
+ "step": 30
52
+ },
53
+ {
54
+ "epoch": 1.1666666666666667,
55
+ "grad_norm": 4.460363388061523,
56
+ "learning_rate": 9.991540791356342e-06,
57
+ "loss": 0.4227,
58
+ "step": 35
59
+ },
60
+ {
61
+ "epoch": 1.3333333333333333,
62
+ "grad_norm": 5.932071685791016,
63
+ "learning_rate": 9.966191788709716e-06,
64
+ "loss": 0.3738,
65
+ "step": 40
66
+ },
67
+ {
68
+ "epoch": 1.5,
69
+ "grad_norm": 5.956723690032959,
70
+ "learning_rate": 9.924038765061042e-06,
71
+ "loss": 0.3456,
72
+ "step": 45
73
+ },
74
+ {
75
+ "epoch": 1.6666666666666665,
76
+ "grad_norm": 3.7256667613983154,
77
+ "learning_rate": 9.86522435289912e-06,
78
+ "loss": 0.3267,
79
+ "step": 50
80
+ },
81
+ {
82
+ "epoch": 1.8333333333333335,
83
+ "grad_norm": 5.209020137786865,
84
+ "learning_rate": 9.789947561577445e-06,
85
+ "loss": 0.3079,
86
+ "step": 55
87
+ },
88
+ {
89
+ "epoch": 2.0,
90
+ "grad_norm": 10.075045585632324,
91
+ "learning_rate": 9.698463103929542e-06,
92
+ "loss": 0.2969,
93
+ "step": 60
94
+ },
95
+ {
96
+ "epoch": 2.1666666666666665,
97
+ "grad_norm": 4.17183256149292,
98
+ "learning_rate": 9.591080534401371e-06,
99
+ "loss": 0.2805,
100
+ "step": 65
101
+ },
102
+ {
103
+ "epoch": 2.3333333333333335,
104
+ "grad_norm": 4.0775065422058105,
105
+ "learning_rate": 9.468163201617063e-06,
106
+ "loss": 0.257,
107
+ "step": 70
108
+ },
109
+ {
110
+ "epoch": 2.5,
111
+ "grad_norm": 6.453698635101318,
112
+ "learning_rate": 9.330127018922195e-06,
113
+ "loss": 0.2403,
114
+ "step": 75
115
+ },
116
+ {
117
+ "epoch": 2.6666666666666665,
118
+ "grad_norm": 5.133937835693359,
119
+ "learning_rate": 9.177439057064684e-06,
120
+ "loss": 0.2375,
121
+ "step": 80
122
+ },
123
+ {
124
+ "epoch": 2.8333333333333335,
125
+ "grad_norm": 5.349823474884033,
126
+ "learning_rate": 9.01061596377522e-06,
127
+ "loss": 0.2512,
128
+ "step": 85
129
+ },
130
+ {
131
+ "epoch": 3.0,
132
+ "grad_norm": 8.886929512023926,
133
+ "learning_rate": 8.83022221559489e-06,
134
+ "loss": 0.2365,
135
+ "step": 90
136
+ },
137
+ {
138
+ "epoch": 3.1666666666666665,
139
+ "grad_norm": 5.774188995361328,
140
+ "learning_rate": 8.636868207865244e-06,
141
+ "loss": 0.2252,
142
+ "step": 95
143
+ },
144
+ {
145
+ "epoch": 3.3333333333333335,
146
+ "grad_norm": 6.560914993286133,
147
+ "learning_rate": 8.43120818934367e-06,
148
+ "loss": 0.2158,
149
+ "step": 100
150
+ },
151
+ {
152
+ "epoch": 3.5,
153
+ "grad_norm": 12.387423515319824,
154
+ "learning_rate": 8.213938048432697e-06,
155
+ "loss": 0.2081,
156
+ "step": 105
157
+ },
158
+ {
159
+ "epoch": 3.6666666666666665,
160
+ "grad_norm": 9.847865104675293,
161
+ "learning_rate": 7.985792958513932e-06,
162
+ "loss": 0.2164,
163
+ "step": 110
164
+ },
165
+ {
166
+ "epoch": 3.8333333333333335,
167
+ "grad_norm": 7.43393087387085,
168
+ "learning_rate": 7.747544890354031e-06,
169
+ "loss": 0.2104,
170
+ "step": 115
171
+ },
172
+ {
173
+ "epoch": 4.0,
174
+ "grad_norm": 6.650928020477295,
175
+ "learning_rate": 7.500000000000001e-06,
176
+ "loss": 0.2011,
177
+ "step": 120
178
+ },
179
+ {
180
+ "epoch": 4.166666666666667,
181
+ "grad_norm": 5.313305854797363,
182
+ "learning_rate": 7.243995901002312e-06,
183
+ "loss": 0.1839,
184
+ "step": 125
185
+ },
186
+ {
187
+ "epoch": 4.333333333333333,
188
+ "grad_norm": 9.173078536987305,
189
+ "learning_rate": 6.980398830195785e-06,
190
+ "loss": 0.1853,
191
+ "step": 130
192
+ },
193
+ {
194
+ "epoch": 4.5,
195
+ "grad_norm": 5.460617542266846,
196
+ "learning_rate": 6.710100716628345e-06,
197
+ "loss": 0.1922,
198
+ "step": 135
199
+ },
200
+ {
201
+ "epoch": 4.666666666666667,
202
+ "grad_norm": 9.463379859924316,
203
+ "learning_rate": 6.434016163555452e-06,
204
+ "loss": 0.2092,
205
+ "step": 140
206
+ },
207
+ {
208
+ "epoch": 4.833333333333333,
209
+ "grad_norm": 7.925484657287598,
210
+ "learning_rate": 6.153079353712201e-06,
211
+ "loss": 0.197,
212
+ "step": 145
213
+ },
214
+ {
215
+ "epoch": 5.0,
216
+ "grad_norm": 8.87891674041748,
217
+ "learning_rate": 5.8682408883346535e-06,
218
+ "loss": 0.1899,
219
+ "step": 150
220
+ },
221
+ {
222
+ "epoch": 5.166666666666667,
223
+ "grad_norm": 6.015185356140137,
224
+ "learning_rate": 5.5804645706261515e-06,
225
+ "loss": 0.1744,
226
+ "step": 155
227
+ },
228
+ {
229
+ "epoch": 5.333333333333333,
230
+ "grad_norm": 6.5013227462768555,
231
+ "learning_rate": 5.290724144552379e-06,
232
+ "loss": 0.1733,
233
+ "step": 160
234
+ },
235
+ {
236
+ "epoch": 5.5,
237
+ "grad_norm": 3.8927905559539795,
238
+ "learning_rate": 5e-06,
239
+ "loss": 0.1665,
240
+ "step": 165
241
+ },
242
+ {
243
+ "epoch": 5.666666666666667,
244
+ "grad_norm": 7.750868797302246,
245
+ "learning_rate": 4.7092758554476215e-06,
246
+ "loss": 0.1695,
247
+ "step": 170
248
+ },
249
+ {
250
+ "epoch": 5.833333333333333,
251
+ "grad_norm": 8.282203674316406,
252
+ "learning_rate": 4.4195354293738484e-06,
253
+ "loss": 0.1729,
254
+ "step": 175
255
+ },
256
+ {
257
+ "epoch": 6.0,
258
+ "grad_norm": 5.089918613433838,
259
+ "learning_rate": 4.131759111665349e-06,
260
+ "loss": 0.1677,
261
+ "step": 180
262
+ },
263
+ {
264
+ "epoch": 6.166666666666667,
265
+ "grad_norm": 5.203615188598633,
266
+ "learning_rate": 3.8469206462878e-06,
267
+ "loss": 0.1616,
268
+ "step": 185
269
+ },
270
+ {
271
+ "epoch": 6.333333333333333,
272
+ "grad_norm": 10.215794563293457,
273
+ "learning_rate": 3.5659838364445505e-06,
274
+ "loss": 0.1631,
275
+ "step": 190
276
+ },
277
+ {
278
+ "epoch": 6.5,
279
+ "grad_norm": 7.743990898132324,
280
+ "learning_rate": 3.289899283371657e-06,
281
+ "loss": 0.1627,
282
+ "step": 195
283
+ },
284
+ {
285
+ "epoch": 6.666666666666667,
286
+ "grad_norm": 5.7741546630859375,
287
+ "learning_rate": 3.019601169804216e-06,
288
+ "loss": 0.16,
289
+ "step": 200
290
+ },
291
+ {
292
+ "epoch": 6.833333333333333,
293
+ "grad_norm": 7.476262092590332,
294
+ "learning_rate": 2.7560040989976894e-06,
295
+ "loss": 0.1543,
296
+ "step": 205
297
+ },
298
+ {
299
+ "epoch": 7.0,
300
+ "grad_norm": 6.046145915985107,
301
+ "learning_rate": 2.5000000000000015e-06,
302
+ "loss": 0.1547,
303
+ "step": 210
304
+ },
305
+ {
306
+ "epoch": 7.166666666666667,
307
+ "grad_norm": 12.649035453796387,
308
+ "learning_rate": 2.2524551096459703e-06,
309
+ "loss": 0.1432,
310
+ "step": 215
311
+ },
312
+ {
313
+ "epoch": 7.333333333333333,
314
+ "grad_norm": 10.913522720336914,
315
+ "learning_rate": 2.0142070414860704e-06,
316
+ "loss": 0.1427,
317
+ "step": 220
318
+ },
319
+ {
320
+ "epoch": 7.5,
321
+ "grad_norm": 5.837172985076904,
322
+ "learning_rate": 1.7860619515673034e-06,
323
+ "loss": 0.1426,
324
+ "step": 225
325
+ },
326
+ {
327
+ "epoch": 7.666666666666667,
328
+ "grad_norm": 13.21788215637207,
329
+ "learning_rate": 1.5687918106563326e-06,
330
+ "loss": 0.1431,
331
+ "step": 230
332
+ },
333
+ {
334
+ "epoch": 7.833333333333333,
335
+ "grad_norm": 4.8505120277404785,
336
+ "learning_rate": 1.3631317921347564e-06,
337
+ "loss": 0.1411,
338
+ "step": 235
339
+ },
340
+ {
341
+ "epoch": 8.0,
342
+ "grad_norm": 4.8936448097229,
343
+ "learning_rate": 1.1697777844051105e-06,
344
+ "loss": 0.1379,
345
+ "step": 240
346
+ },
347
+ {
348
+ "epoch": 8.166666666666666,
349
+ "grad_norm": 6.43206787109375,
350
+ "learning_rate": 9.893840362247809e-07,
351
+ "loss": 0.133,
352
+ "step": 245
353
+ },
354
+ {
355
+ "epoch": 8.333333333333334,
356
+ "grad_norm": 7.297297477722168,
357
+ "learning_rate": 8.225609429353187e-07,
358
+ "loss": 0.1303,
359
+ "step": 250
360
+ },
361
+ {
362
+ "epoch": 8.5,
363
+ "grad_norm": 3.71684193611145,
364
+ "learning_rate": 6.698729810778065e-07,
365
+ "loss": 0.1266,
366
+ "step": 255
367
+ },
368
+ {
369
+ "epoch": 8.666666666666666,
370
+ "grad_norm": 5.448196887969971,
371
+ "learning_rate": 5.318367983829393e-07,
372
+ "loss": 0.1294,
373
+ "step": 260
374
+ },
375
+ {
376
+ "epoch": 8.833333333333334,
377
+ "grad_norm": 4.3221964836120605,
378
+ "learning_rate": 4.089194655986306e-07,
379
+ "loss": 0.1276,
380
+ "step": 265
381
+ },
382
+ {
383
+ "epoch": 9.0,
384
+ "grad_norm": 9.115724563598633,
385
+ "learning_rate": 3.015368960704584e-07,
386
+ "loss": 0.1269,
387
+ "step": 270
388
+ },
389
+ {
390
+ "epoch": 9.166666666666666,
391
+ "grad_norm": 6.583105564117432,
392
+ "learning_rate": 2.1005243842255552e-07,
393
+ "loss": 0.1212,
394
+ "step": 275
395
+ },
396
+ {
397
+ "epoch": 9.333333333333334,
398
+ "grad_norm": 7.882899761199951,
399
+ "learning_rate": 1.3477564710088097e-07,
400
+ "loss": 0.1216,
401
+ "step": 280
402
+ },
403
+ {
404
+ "epoch": 9.5,
405
+ "grad_norm": 5.73756217956543,
406
+ "learning_rate": 7.59612349389599e-08,
407
+ "loss": 0.1206,
408
+ "step": 285
409
+ },
410
+ {
411
+ "epoch": 9.666666666666666,
412
+ "grad_norm": 4.246568202972412,
413
+ "learning_rate": 3.3808211290284886e-08,
414
+ "loss": 0.1221,
415
+ "step": 290
416
+ },
417
+ {
418
+ "epoch": 9.833333333333334,
419
+ "grad_norm": 5.009049415588379,
420
+ "learning_rate": 8.459208643659122e-09,
421
+ "loss": 0.1201,
422
+ "step": 295
423
+ },
424
+ {
425
+ "epoch": 10.0,
426
+ "grad_norm": 5.221693515777588,
427
+ "learning_rate": 0.0,
428
+ "loss": 0.1227,
429
+ "step": 300
430
+ },
431
+ {
432
+ "epoch": 10.0,
433
+ "step": 300,
434
+ "total_flos": 8.00754383150121e+16,
435
+ "train_loss": 0.29854341844717663,
436
+ "train_runtime": 5255.3389,
437
+ "train_samples_per_second": 3.045,
438
+ "train_steps_per_second": 0.057
439
+ }
440
+ ],
441
+ "logging_steps": 5,
442
+ "max_steps": 300,
443
+ "num_input_tokens_seen": 0,
444
+ "num_train_epochs": 10,
445
+ "save_steps": 500,
446
+ "stateful_callbacks": {
447
+ "TrainerControl": {
448
+ "args": {
449
+ "should_epoch_stop": false,
450
+ "should_evaluate": false,
451
+ "should_log": false,
452
+ "should_save": false,
453
+ "should_training_stop": false
454
+ },
455
+ "attributes": {}
456
+ }
457
+ },
458
+ "total_flos": 8.00754383150121e+16,
459
+ "train_batch_size": 18,
460
+ "trial_name": null,
461
+ "trial_params": null
462
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bfa0af299a07a0c2516a41cbe30d1bdac43b650378a7cc37cd9f482feb819a5
3
+ size 7224
vocab.json ADDED
The diff for this file is too large to render. See raw diff