sidbaines commited on
Commit
60e0dae
·
verified ·
1 Parent(s): 20a5661

4b scale-up charter SFT checkpoint 48

Browse files
.gitattributes CHANGED
@@ -57,3 +57,4 @@ sft_4epoch/charter/checkpoint-4/tokenizer.json filter=lfs diff=lfs merge=lfs -te
57
  sft_4epoch/charter/checkpoint-12/tokenizer.json filter=lfs diff=lfs merge=lfs -text
58
  sft_4epoch/charter/checkpoint-24/tokenizer.json filter=lfs diff=lfs merge=lfs -text
59
  sft_4epoch/charter/checkpoint-36/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
57
  sft_4epoch/charter/checkpoint-12/tokenizer.json filter=lfs diff=lfs merge=lfs -text
58
  sft_4epoch/charter/checkpoint-24/tokenizer.json filter=lfs diff=lfs merge=lfs -text
59
  sft_4epoch/charter/checkpoint-36/tokenizer.json filter=lfs diff=lfs merge=lfs -text
60
+ sft_4epoch/charter/checkpoint-48/tokenizer.json filter=lfs diff=lfs merge=lfs -text
sft_4epoch/charter/checkpoint-48/chat_template.jinja ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{ bos_token }}
2
+ {%- if messages[0]['role'] == 'system' -%}
3
+ {%- if messages[0]['content'] is string -%}
4
+ {%- set first_user_prefix = messages[0]['content'] + '
5
+
6
+ ' -%}
7
+ {%- else -%}
8
+ {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
9
+
10
+ ' -%}
11
+ {%- endif -%}
12
+ {%- set loop_messages = messages[1:] -%}
13
+ {%- else -%}
14
+ {%- set first_user_prefix = "" -%}
15
+ {%- set loop_messages = messages -%}
16
+ {%- endif -%}
17
+ {%- for message in loop_messages -%}
18
+ {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
19
+ {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
20
+ {%- endif -%}
21
+ {%- if (message['role'] == 'assistant') -%}
22
+ {%- set role = "model" -%}
23
+ {%- else -%}
24
+ {%- set role = message['role'] -%}
25
+ {%- endif -%}
26
+ {{ '<start_of_turn>' + role + '
27
+ ' + (first_user_prefix if loop.first else "") }}
28
+ {%- if message['content'] is string -%}
29
+ {{ message['content'] | trim }}
30
+ {%- elif message['content'] is iterable -%}
31
+ {%- for item in message['content'] -%}
32
+ {%- if item['type'] == 'image' -%}
33
+ {{ '<start_of_image>' }}
34
+ {%- elif item['type'] == 'text' -%}
35
+ {{ item['text'] | trim }}
36
+ {%- endif -%}
37
+ {%- endfor -%}
38
+ {%- else -%}
39
+ {{ raise_exception("Invalid content type") }}
40
+ {%- endif -%}
41
+ {{ '<end_of_turn>
42
+ ' }}
43
+ {%- endfor -%}
44
+ {%- if add_generation_prompt -%}
45
+ {{'<start_of_turn>model
46
+ '}}
47
+ {%- endif -%}
sft_4epoch/charter/checkpoint-48/config.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma3ForConditionalGeneration"
4
+ ],
5
+ "boi_token_index": 255999,
6
+ "bos_token_id": 2,
7
+ "dtype": "bfloat16",
8
+ "eoi_token_index": 256000,
9
+ "eos_token_id": 1,
10
+ "image_token_index": 262144,
11
+ "initializer_range": 0.02,
12
+ "mm_tokens_per_image": 256,
13
+ "model_type": "gemma3",
14
+ "pad_token_id": 0,
15
+ "text_config": {
16
+ "_sliding_window_pattern": 6,
17
+ "attention_bias": false,
18
+ "attention_dropout": 0.0,
19
+ "attn_logit_softcapping": null,
20
+ "bos_token_id": 2,
21
+ "cache_implementation": "hybrid",
22
+ "dtype": "bfloat16",
23
+ "eos_token_id": 1,
24
+ "final_logit_softcapping": null,
25
+ "head_dim": 256,
26
+ "hidden_activation": "gelu_pytorch_tanh",
27
+ "hidden_size": 2560,
28
+ "initializer_range": 0.02,
29
+ "intermediate_size": 10240,
30
+ "layer_types": [
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "full_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "full_attention",
43
+ "sliding_attention",
44
+ "sliding_attention",
45
+ "sliding_attention",
46
+ "sliding_attention",
47
+ "sliding_attention",
48
+ "full_attention",
49
+ "sliding_attention",
50
+ "sliding_attention",
51
+ "sliding_attention",
52
+ "sliding_attention",
53
+ "sliding_attention",
54
+ "full_attention",
55
+ "sliding_attention",
56
+ "sliding_attention",
57
+ "sliding_attention",
58
+ "sliding_attention",
59
+ "sliding_attention",
60
+ "full_attention",
61
+ "sliding_attention",
62
+ "sliding_attention",
63
+ "sliding_attention",
64
+ "sliding_attention"
65
+ ],
66
+ "max_position_embeddings": 131072,
67
+ "model_type": "gemma3_text",
68
+ "num_attention_heads": 8,
69
+ "num_hidden_layers": 34,
70
+ "num_key_value_heads": 4,
71
+ "pad_token_id": 0,
72
+ "query_pre_attn_scalar": 256,
73
+ "rms_norm_eps": 1e-06,
74
+ "rope_parameters": {
75
+ "full_attention": {
76
+ "factor": 8.0,
77
+ "rope_theta": 1000000.0,
78
+ "rope_type": "linear"
79
+ },
80
+ "sliding_attention": {
81
+ "rope_theta": 10000.0,
82
+ "rope_type": "default"
83
+ }
84
+ },
85
+ "sliding_window": 1024,
86
+ "sliding_window_pattern": 6,
87
+ "tie_word_embeddings": true,
88
+ "use_bidirectional_attention": false,
89
+ "use_cache": false,
90
+ "vocab_size": 262208
91
+ },
92
+ "tie_word_embeddings": true,
93
+ "transformers_version": "5.9.0",
94
+ "unsloth_fixed": true,
95
+ "use_cache": false,
96
+ "vision_config": {
97
+ "attention_dropout": 0.0,
98
+ "dtype": "bfloat16",
99
+ "hidden_act": "gelu_pytorch_tanh",
100
+ "hidden_size": 1152,
101
+ "image_size": 896,
102
+ "intermediate_size": 4304,
103
+ "layer_norm_eps": 1e-06,
104
+ "model_type": "siglip_vision_model",
105
+ "num_attention_heads": 16,
106
+ "num_channels": 3,
107
+ "num_hidden_layers": 27,
108
+ "patch_size": 14,
109
+ "vision_use_head": false
110
+ }
111
+ }
sft_4epoch/charter/checkpoint-48/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "pad_token_id": 0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.9.0"
13
+ }
sft_4epoch/charter/checkpoint-48/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:544b6ec660a37442b519e667b6758ff876fa40f265e1c3069325d106ad9bd679
3
+ size 9942783064
sft_4epoch/charter/checkpoint-48/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4e5e6de705f1f5db543d57b295297c91076ba869a51edcfaaaeed055d811452
3
+ size 15521485995
sft_4epoch/charter/checkpoint-48/preprocessor_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_pan_and_scan": null,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "image_mean": [
8
+ 0.5,
9
+ 0.5,
10
+ 0.5
11
+ ],
12
+ "image_processor_type": "Gemma3ImageProcessor",
13
+ "image_seq_length": 256,
14
+ "image_std": [
15
+ 0.5,
16
+ 0.5,
17
+ 0.5
18
+ ],
19
+ "pan_and_scan_max_num_crops": null,
20
+ "pan_and_scan_min_crop_size": null,
21
+ "pan_and_scan_min_ratio_to_activate": null,
22
+ "processor_class": "Gemma3Processor",
23
+ "resample": 2,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "height": 896,
27
+ "width": 896
28
+ }
29
+ }
sft_4epoch/charter/checkpoint-48/processor_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "image_seq_length": 256,
3
+ "processor_class": "Gemma3Processor"
4
+ }
sft_4epoch/charter/checkpoint-48/pytorch_model_fsdp.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18136af7e1ed98523a8775b99302c5f22f952032e5936ab5db985506a5ab9ac1
3
+ size 9942970813
sft_4epoch/charter/checkpoint-48/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6ae654ed628e46a69a3185a2008a760ecb6fa76a6ed8ea36091fbf51572a9dc
3
+ size 14917
sft_4epoch/charter/checkpoint-48/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca90864f21237f3b2ef7719c284655f031e849a610ff91881a6d9e258037405
3
+ size 14917
sft_4epoch/charter/checkpoint-48/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec3414e5c8b4d3b47de7f61d5314acd64635f9517a355779879e73c2585549d1
3
+ size 1465
sft_4epoch/charter/checkpoint-48/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:daab2354f8a74e70d70b4d1f804939b68a8c9624dd06cb7858e52dd8970e9726
3
+ size 33384567
sft_4epoch/charter/checkpoint-48/tokenizer_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "boi_token": "<start_of_image>",
4
+ "bos_token": "<bos>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eoi_token": "<end_of_image>",
7
+ "eos_token": "<eos>",
8
+ "image_token": "<image_soft_token>",
9
+ "is_local": true,
10
+ "local_files_only": false,
11
+ "mask_token": "<mask>",
12
+ "model_max_length": 131072,
13
+ "model_specific_special_tokens": {
14
+ "boi_token": "<start_of_image>",
15
+ "eoi_token": "<end_of_image>",
16
+ "image_token": "<image_soft_token>"
17
+ },
18
+ "pad_token": "<pad>",
19
+ "padding_side": "left",
20
+ "processor_class": "Gemma3Processor",
21
+ "sp_model_kwargs": null,
22
+ "spaces_between_special_tokens": false,
23
+ "tokenizer_class": "GemmaTokenizer",
24
+ "unk_token": "<unk>",
25
+ "use_default_system_prompt": false
26
+ }
sft_4epoch/charter/checkpoint-48/tokens_state.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"total": 100638720, "trainable": 62647352}
sft_4epoch/charter/checkpoint-48/trainer_state.json ADDED
@@ -0,0 +1,706 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08533333333333333,
6
+ "eval_steps": 500,
7
+ "global_step": 48,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.0017777777777777779,
14
+ "grad_norm": 2.9375,
15
+ "learning_rate": 0.0,
16
+ "loss": 1.1085205078125,
17
+ "memory/device_reserved (GiB)": 40.7,
18
+ "memory/max_active (GiB)": 34.07,
19
+ "memory/max_allocated (GiB)": 34.07,
20
+ "ppl": 3.02987,
21
+ "step": 1,
22
+ "tokens/total": 2097152,
23
+ "tokens/train_per_sec_per_gpu": 614.88,
24
+ "tokens/trainable": 1319329
25
+ },
26
+ {
27
+ "epoch": 0.0035555555555555557,
28
+ "grad_norm": 2.921875,
29
+ "learning_rate": 3.3333333333333333e-06,
30
+ "loss": 1.093994140625,
31
+ "memory/device_reserved (GiB)": 47.77,
32
+ "memory/max_active (GiB)": 41.3,
33
+ "memory/max_allocated (GiB)": 41.3,
34
+ "ppl": 2.98618,
35
+ "step": 2,
36
+ "tokens/total": 4194304,
37
+ "tokens/train_per_sec_per_gpu": 617.46,
38
+ "tokens/trainable": 2628278
39
+ },
40
+ {
41
+ "epoch": 0.005333333333333333,
42
+ "grad_norm": 2.609375,
43
+ "learning_rate": 6.666666666666667e-06,
44
+ "loss": 1.0843505859375,
45
+ "memory/device_reserved (GiB)": 47.77,
46
+ "memory/max_active (GiB)": 41.3,
47
+ "memory/max_allocated (GiB)": 41.3,
48
+ "ppl": 2.95752,
49
+ "step": 3,
50
+ "tokens/total": 6291456,
51
+ "tokens/train_per_sec_per_gpu": 655.41,
52
+ "tokens/trainable": 3948806
53
+ },
54
+ {
55
+ "epoch": 0.0071111111111111115,
56
+ "grad_norm": 1.7421875,
57
+ "learning_rate": 1e-05,
58
+ "loss": 1.0223388671875,
59
+ "memory/device_reserved (GiB)": 47.77,
60
+ "memory/max_active (GiB)": 41.3,
61
+ "memory/max_allocated (GiB)": 41.3,
62
+ "ppl": 2.77969,
63
+ "step": 4,
64
+ "tokens/total": 8388608,
65
+ "tokens/train_per_sec_per_gpu": 598.2,
66
+ "tokens/trainable": 5229996
67
+ },
68
+ {
69
+ "epoch": 0.008888888888888889,
70
+ "grad_norm": 2.03125,
71
+ "learning_rate": 9.98903822616921e-06,
72
+ "loss": 1.0355224609375,
73
+ "memory/device_reserved (GiB)": 47.8,
74
+ "memory/max_active (GiB)": 41.3,
75
+ "memory/max_allocated (GiB)": 41.3,
76
+ "ppl": 2.81658,
77
+ "step": 5,
78
+ "tokens/total": 10485760,
79
+ "tokens/train_per_sec_per_gpu": 684.37,
80
+ "tokens/trainable": 6541073
81
+ },
82
+ {
83
+ "epoch": 0.010666666666666666,
84
+ "grad_norm": 1.703125,
85
+ "learning_rate": 9.956206309337067e-06,
86
+ "loss": 1.012939453125,
87
+ "memory/device_reserved (GiB)": 47.82,
88
+ "memory/max_active (GiB)": 41.3,
89
+ "memory/max_allocated (GiB)": 41.3,
90
+ "ppl": 2.75368,
91
+ "step": 6,
92
+ "tokens/total": 12582912,
93
+ "tokens/train_per_sec_per_gpu": 634.23,
94
+ "tokens/trainable": 7870413
95
+ },
96
+ {
97
+ "epoch": 0.012444444444444444,
98
+ "grad_norm": 1.28125,
99
+ "learning_rate": 9.901664203302126e-06,
100
+ "loss": 0.9525146484375,
101
+ "memory/device_reserved (GiB)": 47.82,
102
+ "memory/max_active (GiB)": 41.3,
103
+ "memory/max_allocated (GiB)": 41.3,
104
+ "ppl": 2.59222,
105
+ "step": 7,
106
+ "tokens/total": 14680064,
107
+ "tokens/train_per_sec_per_gpu": 635.02,
108
+ "tokens/trainable": 9189777
109
+ },
110
+ {
111
+ "epoch": 0.014222222222222223,
112
+ "grad_norm": 0.91015625,
113
+ "learning_rate": 9.825677631722436e-06,
114
+ "loss": 0.9556884765625,
115
+ "memory/device_reserved (GiB)": 47.82,
116
+ "memory/max_active (GiB)": 41.3,
117
+ "memory/max_allocated (GiB)": 41.3,
118
+ "ppl": 2.60046,
119
+ "step": 8,
120
+ "tokens/total": 16777216,
121
+ "tokens/train_per_sec_per_gpu": 618.16,
122
+ "tokens/trainable": 10479490
123
+ },
124
+ {
125
+ "epoch": 0.016,
126
+ "grad_norm": 0.85546875,
127
+ "learning_rate": 9.728616793536588e-06,
128
+ "loss": 0.9462890625,
129
+ "memory/device_reserved (GiB)": 47.82,
130
+ "memory/max_active (GiB)": 41.3,
131
+ "memory/max_allocated (GiB)": 41.3,
132
+ "ppl": 2.57613,
133
+ "step": 9,
134
+ "tokens/total": 18874368,
135
+ "tokens/train_per_sec_per_gpu": 652.09,
136
+ "tokens/trainable": 11784999
137
+ },
138
+ {
139
+ "epoch": 0.017777777777777778,
140
+ "grad_norm": 0.67578125,
141
+ "learning_rate": 9.610954559391704e-06,
142
+ "loss": 0.958740234375,
143
+ "memory/device_reserved (GiB)": 47.82,
144
+ "memory/max_active (GiB)": 41.3,
145
+ "memory/max_allocated (GiB)": 41.3,
146
+ "ppl": 2.60841,
147
+ "step": 10,
148
+ "tokens/total": 20971520,
149
+ "tokens/train_per_sec_per_gpu": 600.22,
150
+ "tokens/trainable": 13094236
151
+ },
152
+ {
153
+ "epoch": 0.019555555555555555,
154
+ "grad_norm": 0.87109375,
155
+ "learning_rate": 9.473264167865172e-06,
156
+ "loss": 0.943359375,
157
+ "memory/device_reserved (GiB)": 47.82,
158
+ "memory/max_active (GiB)": 41.3,
159
+ "memory/max_allocated (GiB)": 41.3,
160
+ "ppl": 2.5686,
161
+ "step": 11,
162
+ "tokens/total": 23068672,
163
+ "tokens/train_per_sec_per_gpu": 610.29,
164
+ "tokens/trainable": 14404718
165
+ },
166
+ {
167
+ "epoch": 0.021333333333333333,
168
+ "grad_norm": 0.6640625,
169
+ "learning_rate": 9.316216432703918e-06,
170
+ "loss": 0.9000244140625,
171
+ "memory/device_reserved (GiB)": 47.82,
172
+ "memory/max_active (GiB)": 41.3,
173
+ "memory/max_allocated (GiB)": 41.3,
174
+ "ppl": 2.45966,
175
+ "step": 12,
176
+ "tokens/total": 25165824,
177
+ "tokens/train_per_sec_per_gpu": 667.87,
178
+ "tokens/trainable": 15735257
179
+ },
180
+ {
181
+ "epoch": 0.02311111111111111,
182
+ "grad_norm": 0.55859375,
183
+ "learning_rate": 9.140576474687263e-06,
184
+ "loss": 0.94091796875,
185
+ "memory/device_reserved (GiB)": 47.82,
186
+ "memory/max_active (GiB)": 41.3,
187
+ "memory/max_allocated (GiB)": 41.3,
188
+ "ppl": 2.56233,
189
+ "step": 13,
190
+ "tokens/total": 27262976,
191
+ "tokens/train_per_sec_per_gpu": 634.12,
192
+ "tokens/trainable": 17038230
193
+ },
194
+ {
195
+ "epoch": 0.024888888888888887,
196
+ "grad_norm": 0.76953125,
197
+ "learning_rate": 8.947199994035402e-06,
198
+ "loss": 0.921875,
199
+ "memory/device_reserved (GiB)": 47.82,
200
+ "memory/max_active (GiB)": 41.3,
201
+ "memory/max_allocated (GiB)": 41.3,
202
+ "ppl": 2.514,
203
+ "step": 14,
204
+ "tokens/total": 29360128,
205
+ "tokens/train_per_sec_per_gpu": 705.62,
206
+ "tokens/trainable": 18338536
207
+ },
208
+ {
209
+ "epoch": 0.02666666666666667,
210
+ "grad_norm": 0.9140625,
211
+ "learning_rate": 8.737029101523931e-06,
212
+ "loss": 0.9312744140625,
213
+ "memory/device_reserved (GiB)": 47.82,
214
+ "memory/max_active (GiB)": 41.3,
215
+ "memory/max_allocated (GiB)": 41.3,
216
+ "ppl": 2.53774,
217
+ "step": 15,
218
+ "tokens/total": 31457280,
219
+ "tokens/train_per_sec_per_gpu": 587.74,
220
+ "tokens/trainable": 19638784
221
+ },
222
+ {
223
+ "epoch": 0.028444444444444446,
224
+ "grad_norm": 0.494140625,
225
+ "learning_rate": 8.511087728614863e-06,
226
+ "loss": 0.9188232421875,
227
+ "memory/device_reserved (GiB)": 47.82,
228
+ "memory/max_active (GiB)": 41.3,
229
+ "memory/max_allocated (GiB)": 41.3,
230
+ "ppl": 2.50634,
231
+ "step": 16,
232
+ "tokens/total": 33554432,
233
+ "tokens/train_per_sec_per_gpu": 625.92,
234
+ "tokens/trainable": 20962100
235
+ },
236
+ {
237
+ "epoch": 0.030222222222222223,
238
+ "grad_norm": 0.42578125,
239
+ "learning_rate": 8.270476638965463e-06,
240
+ "loss": 0.9246826171875,
241
+ "memory/device_reserved (GiB)": 47.82,
242
+ "memory/max_active (GiB)": 41.3,
243
+ "memory/max_allocated (GiB)": 41.3,
244
+ "ppl": 2.52107,
245
+ "step": 17,
246
+ "tokens/total": 35651584,
247
+ "tokens/train_per_sec_per_gpu": 647.08,
248
+ "tokens/trainable": 22244548
249
+ },
250
+ {
251
+ "epoch": 0.032,
252
+ "grad_norm": 0.5390625,
253
+ "learning_rate": 8.016368065618361e-06,
254
+ "loss": 0.92431640625,
255
+ "memory/device_reserved (GiB)": 47.82,
256
+ "memory/max_active (GiB)": 41.3,
257
+ "memory/max_allocated (GiB)": 41.3,
258
+ "ppl": 2.52014,
259
+ "step": 18,
260
+ "tokens/total": 37748736,
261
+ "tokens/train_per_sec_per_gpu": 633.07,
262
+ "tokens/trainable": 23550208
263
+ },
264
+ {
265
+ "epoch": 0.033777777777777775,
266
+ "grad_norm": 0.439453125,
267
+ "learning_rate": 7.75e-06,
268
+ "loss": 0.9212646484375,
269
+ "memory/device_reserved (GiB)": 47.82,
270
+ "memory/max_active (GiB)": 41.3,
271
+ "memory/max_allocated (GiB)": 41.3,
272
+ "ppl": 2.51247,
273
+ "step": 19,
274
+ "tokens/total": 39845888,
275
+ "tokens/train_per_sec_per_gpu": 654.96,
276
+ "tokens/trainable": 24856052
277
+ },
278
+ {
279
+ "epoch": 0.035555555555555556,
280
+ "grad_norm": 0.419921875,
281
+ "learning_rate": 7.472670160550849e-06,
282
+ "loss": 0.9345703125,
283
+ "memory/device_reserved (GiB)": 47.82,
284
+ "memory/max_active (GiB)": 41.3,
285
+ "memory/max_allocated (GiB)": 41.3,
286
+ "ppl": 2.54612,
287
+ "step": 20,
288
+ "tokens/total": 41943040,
289
+ "tokens/train_per_sec_per_gpu": 613.91,
290
+ "tokens/trainable": 26140298
291
+ },
292
+ {
293
+ "epoch": 0.037333333333333336,
294
+ "grad_norm": 0.58984375,
295
+ "learning_rate": 7.185729670371605e-06,
296
+ "loss": 0.937255859375,
297
+ "memory/device_reserved (GiB)": 47.82,
298
+ "memory/max_active (GiB)": 41.3,
299
+ "memory/max_allocated (GiB)": 41.3,
300
+ "ppl": 2.55297,
301
+ "step": 21,
302
+ "tokens/total": 44040192,
303
+ "tokens/train_per_sec_per_gpu": 627.12,
304
+ "tokens/trainable": 27452072
305
+ },
306
+ {
307
+ "epoch": 0.03911111111111111,
308
+ "grad_norm": 0.435546875,
309
+ "learning_rate": 6.890576474687264e-06,
310
+ "loss": 0.938232421875,
311
+ "memory/device_reserved (GiB)": 47.82,
312
+ "memory/max_active (GiB)": 41.3,
313
+ "memory/max_allocated (GiB)": 41.3,
314
+ "ppl": 2.55546,
315
+ "step": 22,
316
+ "tokens/total": 46137344,
317
+ "tokens/train_per_sec_per_gpu": 640.7,
318
+ "tokens/trainable": 28743186
319
+ },
320
+ {
321
+ "epoch": 0.04088888888888889,
322
+ "grad_norm": 0.41796875,
323
+ "learning_rate": 6.588648530198505e-06,
324
+ "loss": 0.9014892578125,
325
+ "memory/device_reserved (GiB)": 47.82,
326
+ "memory/max_active (GiB)": 41.3,
327
+ "memory/max_allocated (GiB)": 41.3,
328
+ "ppl": 2.46327,
329
+ "step": 23,
330
+ "tokens/total": 48234496,
331
+ "tokens/train_per_sec_per_gpu": 595.72,
332
+ "tokens/trainable": 30066808
333
+ },
334
+ {
335
+ "epoch": 0.042666666666666665,
336
+ "grad_norm": 0.443359375,
337
+ "learning_rate": 6.281416799501188e-06,
338
+ "loss": 0.9259033203125,
339
+ "memory/device_reserved (GiB)": 47.82,
340
+ "memory/max_active (GiB)": 41.3,
341
+ "memory/max_allocated (GiB)": 41.3,
342
+ "ppl": 2.52415,
343
+ "step": 24,
344
+ "tokens/total": 50331648,
345
+ "tokens/train_per_sec_per_gpu": 594.85,
346
+ "tokens/trainable": 31368300
347
+ },
348
+ {
349
+ "epoch": 0.044444444444444446,
350
+ "grad_norm": 0.3984375,
351
+ "learning_rate": 5.970378084704441e-06,
352
+ "loss": 0.9315185546875,
353
+ "memory/device_reserved (GiB)": 47.82,
354
+ "memory/max_active (GiB)": 41.3,
355
+ "memory/max_allocated (GiB)": 41.3,
356
+ "ppl": 2.53836,
357
+ "step": 25,
358
+ "tokens/total": 52428800,
359
+ "tokens/train_per_sec_per_gpu": 622.74,
360
+ "tokens/trainable": 32679264
361
+ },
362
+ {
363
+ "epoch": 0.04622222222222222,
364
+ "grad_norm": 0.375,
365
+ "learning_rate": 5.657047735161256e-06,
366
+ "loss": 0.9356689453125,
367
+ "memory/device_reserved (GiB)": 47.82,
368
+ "memory/max_active (GiB)": 41.3,
369
+ "memory/max_allocated (GiB)": 41.3,
370
+ "ppl": 2.54892,
371
+ "step": 26,
372
+ "tokens/total": 54525952,
373
+ "tokens/train_per_sec_per_gpu": 601.47,
374
+ "tokens/trainable": 33988396
375
+ },
376
+ {
377
+ "epoch": 0.048,
378
+ "grad_norm": 0.41796875,
379
+ "learning_rate": 5.342952264838748e-06,
380
+ "loss": 0.920166015625,
381
+ "memory/device_reserved (GiB)": 47.82,
382
+ "memory/max_active (GiB)": 41.3,
383
+ "memory/max_allocated (GiB)": 41.3,
384
+ "ppl": 2.50971,
385
+ "step": 27,
386
+ "tokens/total": 56623104,
387
+ "tokens/train_per_sec_per_gpu": 634.93,
388
+ "tokens/trainable": 35298496
389
+ },
390
+ {
391
+ "epoch": 0.049777777777777775,
392
+ "grad_norm": 0.390625,
393
+ "learning_rate": 5.02962191529556e-06,
394
+ "loss": 0.920654296875,
395
+ "memory/device_reserved (GiB)": 47.82,
396
+ "memory/max_active (GiB)": 41.3,
397
+ "memory/max_allocated (GiB)": 41.3,
398
+ "ppl": 2.51093,
399
+ "step": 28,
400
+ "tokens/total": 58720256,
401
+ "tokens/train_per_sec_per_gpu": 635.98,
402
+ "tokens/trainable": 36597120
403
+ },
404
+ {
405
+ "epoch": 0.051555555555555556,
406
+ "grad_norm": 0.50390625,
407
+ "learning_rate": 4.718583200498814e-06,
408
+ "loss": 0.9144287109375,
409
+ "memory/device_reserved (GiB)": 47.82,
410
+ "memory/max_active (GiB)": 41.3,
411
+ "memory/max_allocated (GiB)": 41.3,
412
+ "ppl": 2.49535,
413
+ "step": 29,
414
+ "tokens/total": 60817408,
415
+ "tokens/train_per_sec_per_gpu": 661.5,
416
+ "tokens/trainable": 37907664
417
+ },
418
+ {
419
+ "epoch": 0.05333333333333334,
420
+ "grad_norm": 0.41015625,
421
+ "learning_rate": 4.4113514698014955e-06,
422
+ "loss": 0.906982421875,
423
+ "memory/device_reserved (GiB)": 47.82,
424
+ "memory/max_active (GiB)": 41.3,
425
+ "memory/max_allocated (GiB)": 41.3,
426
+ "ppl": 2.47684,
427
+ "step": 30,
428
+ "tokens/total": 62889984,
429
+ "tokens/train_per_sec_per_gpu": 612.59,
430
+ "tokens/trainable": 39186144
431
+ },
432
+ {
433
+ "epoch": 0.05511111111111111,
434
+ "grad_norm": 0.349609375,
435
+ "learning_rate": 4.109423525312738e-06,
436
+ "loss": 0.919921875,
437
+ "memory/device_reserved (GiB)": 47.82,
438
+ "memory/max_active (GiB)": 41.3,
439
+ "memory/max_allocated (GiB)": 41.3,
440
+ "ppl": 2.50909,
441
+ "step": 31,
442
+ "tokens/total": 64987136,
443
+ "tokens/train_per_sec_per_gpu": 646.69,
444
+ "tokens/trainable": 40499084
445
+ },
446
+ {
447
+ "epoch": 0.05688888888888889,
448
+ "grad_norm": 0.4375,
449
+ "learning_rate": 3.8142703296283954e-06,
450
+ "loss": 0.9412841796875,
451
+ "memory/device_reserved (GiB)": 47.82,
452
+ "memory/max_active (GiB)": 41.3,
453
+ "memory/max_allocated (GiB)": 41.3,
454
+ "ppl": 2.56327,
455
+ "step": 32,
456
+ "tokens/total": 67084288,
457
+ "tokens/train_per_sec_per_gpu": 578.71,
458
+ "tokens/trainable": 41754008
459
+ },
460
+ {
461
+ "epoch": 0.058666666666666666,
462
+ "grad_norm": 0.353515625,
463
+ "learning_rate": 3.527329839449152e-06,
464
+ "loss": 0.921142578125,
465
+ "memory/device_reserved (GiB)": 47.82,
466
+ "memory/max_active (GiB)": 41.3,
467
+ "memory/max_allocated (GiB)": 41.3,
468
+ "ppl": 2.51216,
469
+ "step": 33,
470
+ "tokens/total": 69181440,
471
+ "tokens/train_per_sec_per_gpu": 645.1,
472
+ "tokens/trainable": 43059700
473
+ },
474
+ {
475
+ "epoch": 0.060444444444444446,
476
+ "grad_norm": 0.35546875,
477
+ "learning_rate": 3.2500000000000015e-06,
478
+ "loss": 0.886962890625,
479
+ "memory/device_reserved (GiB)": 47.82,
480
+ "memory/max_active (GiB)": 41.3,
481
+ "memory/max_allocated (GiB)": 41.3,
482
+ "ppl": 2.42775,
483
+ "step": 34,
484
+ "tokens/total": 71278592,
485
+ "tokens/train_per_sec_per_gpu": 546.77,
486
+ "tokens/trainable": 44340384
487
+ },
488
+ {
489
+ "epoch": 0.06222222222222222,
490
+ "grad_norm": 0.341796875,
491
+ "learning_rate": 2.98363193438164e-06,
492
+ "loss": 0.9200439453125,
493
+ "memory/device_reserved (GiB)": 47.82,
494
+ "memory/max_active (GiB)": 41.3,
495
+ "memory/max_allocated (GiB)": 41.3,
496
+ "ppl": 2.5094,
497
+ "step": 35,
498
+ "tokens/total": 73375744,
499
+ "tokens/train_per_sec_per_gpu": 633.3,
500
+ "tokens/trainable": 45649076
501
+ },
502
+ {
503
+ "epoch": 0.064,
504
+ "grad_norm": 0.380859375,
505
+ "learning_rate": 2.7295233610345384e-06,
506
+ "loss": 0.929443359375,
507
+ "memory/device_reserved (GiB)": 47.82,
508
+ "memory/max_active (GiB)": 41.3,
509
+ "memory/max_allocated (GiB)": 41.3,
510
+ "ppl": 2.5331,
511
+ "step": 36,
512
+ "tokens/total": 75472896,
513
+ "tokens/train_per_sec_per_gpu": 600.88,
514
+ "tokens/trainable": 46969288
515
+ },
516
+ {
517
+ "epoch": 0.06577777777777778,
518
+ "grad_norm": 0.5859375,
519
+ "learning_rate": 2.4889122713851397e-06,
520
+ "loss": 0.9454345703125,
521
+ "memory/device_reserved (GiB)": 47.82,
522
+ "memory/max_active (GiB)": 41.3,
523
+ "memory/max_allocated (GiB)": 41.3,
524
+ "ppl": 2.57393,
525
+ "step": 37,
526
+ "tokens/total": 77570048,
527
+ "tokens/train_per_sec_per_gpu": 660.32,
528
+ "tokens/trainable": 48301204
529
+ },
530
+ {
531
+ "epoch": 0.06755555555555555,
532
+ "grad_norm": 0.34765625,
533
+ "learning_rate": 2.262970898476071e-06,
534
+ "loss": 0.905029296875,
535
+ "memory/device_reserved (GiB)": 47.82,
536
+ "memory/max_active (GiB)": 41.3,
537
+ "memory/max_allocated (GiB)": 41.3,
538
+ "ppl": 2.472,
539
+ "step": 38,
540
+ "tokens/total": 79667200,
541
+ "tokens/train_per_sec_per_gpu": 601.44,
542
+ "tokens/trainable": 49615048
543
+ },
544
+ {
545
+ "epoch": 0.06933333333333333,
546
+ "grad_norm": 0.328125,
547
+ "learning_rate": 2.0528000059646e-06,
548
+ "loss": 0.895751953125,
549
+ "memory/device_reserved (GiB)": 47.82,
550
+ "memory/max_active (GiB)": 41.3,
551
+ "memory/max_allocated (GiB)": 41.3,
552
+ "ppl": 2.44918,
553
+ "step": 39,
554
+ "tokens/total": 81764352,
555
+ "tokens/train_per_sec_per_gpu": 643.88,
556
+ "tokens/trainable": 50907048
557
+ },
558
+ {
559
+ "epoch": 0.07111111111111111,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 1.8594235253127373e-06,
562
+ "loss": 0.9324951171875,
563
+ "memory/device_reserved (GiB)": 47.82,
564
+ "memory/max_active (GiB)": 41.3,
565
+ "memory/max_allocated (GiB)": 41.3,
566
+ "ppl": 2.54084,
567
+ "step": 40,
568
+ "tokens/total": 83861504,
569
+ "tokens/train_per_sec_per_gpu": 604.16,
570
+ "tokens/trainable": 52206344
571
+ },
572
+ {
573
+ "epoch": 0.07288888888888889,
574
+ "grad_norm": 0.349609375,
575
+ "learning_rate": 1.6837835672960834e-06,
576
+ "loss": 0.929931640625,
577
+ "memory/device_reserved (GiB)": 47.82,
578
+ "memory/max_active (GiB)": 41.3,
579
+ "memory/max_allocated (GiB)": 41.3,
580
+ "ppl": 2.53434,
581
+ "step": 41,
582
+ "tokens/total": 85958656,
583
+ "tokens/train_per_sec_per_gpu": 606.12,
584
+ "tokens/trainable": 53511636
585
+ },
586
+ {
587
+ "epoch": 0.07466666666666667,
588
+ "grad_norm": 0.388671875,
589
+ "learning_rate": 1.526735832134829e-06,
590
+ "loss": 0.9239501953125,
591
+ "memory/device_reserved (GiB)": 47.82,
592
+ "memory/max_active (GiB)": 41.3,
593
+ "memory/max_allocated (GiB)": 41.3,
594
+ "ppl": 2.51922,
595
+ "step": 42,
596
+ "tokens/total": 88055808,
597
+ "tokens/train_per_sec_per_gpu": 641.12,
598
+ "tokens/trainable": 54832244
599
+ },
600
+ {
601
+ "epoch": 0.07644444444444444,
602
+ "grad_norm": 0.34375,
603
+ "learning_rate": 1.389045440608296e-06,
604
+ "loss": 0.907958984375,
605
+ "memory/device_reserved (GiB)": 47.82,
606
+ "memory/max_active (GiB)": 41.3,
607
+ "memory/max_allocated (GiB)": 41.3,
608
+ "ppl": 2.47926,
609
+ "step": 43,
610
+ "tokens/total": 90152960,
611
+ "tokens/train_per_sec_per_gpu": 599.45,
612
+ "tokens/trainable": 56163436
613
+ },
614
+ {
615
+ "epoch": 0.07822222222222222,
616
+ "grad_norm": 0.349609375,
617
+ "learning_rate": 1.2713832064634127e-06,
618
+ "loss": 0.9462890625,
619
+ "memory/device_reserved (GiB)": 47.82,
620
+ "memory/max_active (GiB)": 41.3,
621
+ "memory/max_allocated (GiB)": 41.3,
622
+ "ppl": 2.57613,
623
+ "step": 44,
624
+ "tokens/total": 92250112,
625
+ "tokens/train_per_sec_per_gpu": 641.09,
626
+ "tokens/trainable": 57456500
627
+ },
628
+ {
629
+ "epoch": 0.08,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 1.174322368277565e-06,
632
+ "loss": 0.9158935546875,
633
+ "memory/device_reserved (GiB)": 47.82,
634
+ "memory/max_active (GiB)": 41.3,
635
+ "memory/max_allocated (GiB)": 41.3,
636
+ "ppl": 2.49901,
637
+ "step": 45,
638
+ "tokens/total": 94347264,
639
+ "tokens/train_per_sec_per_gpu": 617.32,
640
+ "tokens/trainable": 58768064
641
+ },
642
+ {
643
+ "epoch": 0.08177777777777778,
644
+ "grad_norm": 1.2421875,
645
+ "learning_rate": 1.0983357966978747e-06,
646
+ "loss": 0.9150390625,
647
+ "memory/device_reserved (GiB)": 47.82,
648
+ "memory/max_active (GiB)": 41.3,
649
+ "memory/max_allocated (GiB)": 41.3,
650
+ "ppl": 2.49687,
651
+ "step": 46,
652
+ "tokens/total": 96444416,
653
+ "tokens/train_per_sec_per_gpu": 653.38,
654
+ "tokens/trainable": 60042444
655
+ },
656
+ {
657
+ "epoch": 0.08355555555555555,
658
+ "grad_norm": 0.474609375,
659
+ "learning_rate": 1.0437936906629336e-06,
660
+ "loss": 0.9189453125,
661
+ "memory/device_reserved (GiB)": 47.82,
662
+ "memory/max_active (GiB)": 41.3,
663
+ "memory/max_allocated (GiB)": 41.3,
664
+ "ppl": 2.50665,
665
+ "step": 47,
666
+ "tokens/total": 98541568,
667
+ "tokens/train_per_sec_per_gpu": 605.99,
668
+ "tokens/trainable": 61333816
669
+ },
670
+ {
671
+ "epoch": 0.08533333333333333,
672
+ "grad_norm": 0.515625,
673
+ "learning_rate": 1.0109617738307914e-06,
674
+ "loss": 0.919189453125,
675
+ "memory/device_reserved (GiB)": 47.82,
676
+ "memory/max_active (GiB)": 41.3,
677
+ "memory/max_allocated (GiB)": 41.3,
678
+ "ppl": 2.50726,
679
+ "step": 48,
680
+ "tokens/total": 100638720,
681
+ "tokens/train_per_sec_per_gpu": 617.43,
682
+ "tokens/trainable": 62647352
683
+ }
684
+ ],
685
+ "logging_steps": 1,
686
+ "max_steps": 48,
687
+ "num_input_tokens_seen": 0,
688
+ "num_train_epochs": 1,
689
+ "save_steps": 500,
690
+ "stateful_callbacks": {
691
+ "TrainerControl": {
692
+ "args": {
693
+ "should_epoch_stop": false,
694
+ "should_evaluate": false,
695
+ "should_log": false,
696
+ "should_save": true,
697
+ "should_training_stop": true
698
+ },
699
+ "attributes": {}
700
+ }
701
+ },
702
+ "total_flos": 2.1883537665147535e+18,
703
+ "train_batch_size": 8,
704
+ "trial_name": null,
705
+ "trial_params": null
706
+ }
sft_4epoch/charter/checkpoint-48/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb6dc761c85b168655dd8ebc2a80f811c40b68515cf41dadbdf055ee9c1af7fc
3
+ size 8913