rqadri commited on
Commit
02f89e1
·
verified ·
1 Parent(s): 25d4f28

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -51,3 +51,4 @@ g1_cleanup_code__edit/pre-anneal/500M/tokenizer.json filter=lfs diff=lfs merge=l
51
  g1_cleanup_code__edit/1000M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
52
  g1_cleanup_code/pre-anneal/500M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
53
  g1_cleanup_code__edit/500M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
51
  g1_cleanup_code__edit/1000M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
52
  g1_cleanup_code/pre-anneal/500M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
53
  g1_cleanup_code__edit/500M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
54
+ g1_jupyter_notebook/500M/tokenizer.json filter=lfs diff=lfs merge=lfs -text
g1_jupyter_notebook/500M/.launching ADDED
@@ -0,0 +1 @@
 
 
1
+ 8830938 2026-09-21 04:38:36 x4704c6s4b0n0 636->700
g1_jupyter_notebook/500M/all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.4095962551199532,
3
+ "total_flos": 4821608300544000.0,
4
+ "train_loss": 0.039095867361341206,
5
+ "train_runtime": 2660.3321,
6
+ "train_samples_per_second": 25.26,
7
+ "train_steps_per_second": 0.263
8
+ }
g1_jupyter_notebook/500M/chat_template.jinja ADDED
@@ -0,0 +1 @@
 
 
1
+ {% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ system_message }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ content }}{% elif message['role'] == 'assistant' %}{{ content + '<eos>' }}{% endif %}{% endfor %}
g1_jupyter_notebook/500M/config.json ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma4ForConditionalGeneration"
4
+ ],
5
+ "audio_config": null,
6
+ "audio_token_id": 258881,
7
+ "boa_token_id": 256000,
8
+ "boi_token_id": 255999,
9
+ "bos_token_id": 2,
10
+ "dtype": "bfloat16",
11
+ "eoa_token_id": 258883,
12
+ "eoa_token_index": 258883,
13
+ "eoi_token_id": 258882,
14
+ "eos_token_id": 1,
15
+ "hidden_size": 5376,
16
+ "image_token_id": 258880,
17
+ "initializer_range": 0.02,
18
+ "model_type": "gemma4",
19
+ "pad_token_id": 0,
20
+ "text_config": {
21
+ "attention_bias": false,
22
+ "attention_dropout": 0.0,
23
+ "attention_k_eq_v": true,
24
+ "bos_token_id": 2,
25
+ "dtype": "bfloat16",
26
+ "enable_moe_block": false,
27
+ "eos_token_id": 1,
28
+ "expert_intermediate_size": null,
29
+ "final_logit_softcapping": 30.0,
30
+ "global_head_dim": 512,
31
+ "head_dim": 256,
32
+ "hidden_activation": "gelu_pytorch_tanh",
33
+ "hidden_size": 5376,
34
+ "hidden_size_per_layer_input": 0,
35
+ "initializer_range": 0.02,
36
+ "intermediate_size": 21504,
37
+ "layer_types": [
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "sliding_attention",
43
+ "full_attention",
44
+ "sliding_attention",
45
+ "sliding_attention",
46
+ "sliding_attention",
47
+ "sliding_attention",
48
+ "sliding_attention",
49
+ "full_attention",
50
+ "sliding_attention",
51
+ "sliding_attention",
52
+ "sliding_attention",
53
+ "sliding_attention",
54
+ "sliding_attention",
55
+ "full_attention",
56
+ "sliding_attention",
57
+ "sliding_attention",
58
+ "sliding_attention",
59
+ "sliding_attention",
60
+ "sliding_attention",
61
+ "full_attention",
62
+ "sliding_attention",
63
+ "sliding_attention",
64
+ "sliding_attention",
65
+ "sliding_attention",
66
+ "sliding_attention",
67
+ "full_attention",
68
+ "sliding_attention",
69
+ "sliding_attention",
70
+ "sliding_attention",
71
+ "sliding_attention",
72
+ "sliding_attention",
73
+ "full_attention",
74
+ "sliding_attention",
75
+ "sliding_attention",
76
+ "sliding_attention",
77
+ "sliding_attention",
78
+ "sliding_attention",
79
+ "full_attention",
80
+ "sliding_attention",
81
+ "sliding_attention",
82
+ "sliding_attention",
83
+ "sliding_attention",
84
+ "sliding_attention",
85
+ "full_attention",
86
+ "sliding_attention",
87
+ "sliding_attention",
88
+ "sliding_attention",
89
+ "sliding_attention",
90
+ "sliding_attention",
91
+ "full_attention",
92
+ "sliding_attention",
93
+ "sliding_attention",
94
+ "sliding_attention",
95
+ "sliding_attention",
96
+ "sliding_attention",
97
+ "full_attention"
98
+ ],
99
+ "max_position_embeddings": 262144,
100
+ "model_type": "gemma4_text",
101
+ "moe_intermediate_size": null,
102
+ "num_attention_heads": 32,
103
+ "num_experts": null,
104
+ "num_global_key_value_heads": 4,
105
+ "num_hidden_layers": 60,
106
+ "num_key_value_heads": 16,
107
+ "num_kv_shared_layers": 0,
108
+ "pad_token_id": 0,
109
+ "rms_norm_eps": 1e-06,
110
+ "rope_parameters": {
111
+ "full_attention": {
112
+ "partial_rotary_factor": 0.25,
113
+ "rope_theta": 1000000.0,
114
+ "rope_type": "proportional"
115
+ },
116
+ "sliding_attention": {
117
+ "rope_theta": 10000.0,
118
+ "rope_type": "default"
119
+ }
120
+ },
121
+ "sliding_window": 1024,
122
+ "tie_word_embeddings": true,
123
+ "top_k_experts": null,
124
+ "use_bidirectional_attention": "vision",
125
+ "use_cache": false,
126
+ "use_double_wide_mlp": false,
127
+ "vocab_size": 262144,
128
+ "vocab_size_per_layer_input": 262144
129
+ },
130
+ "tie_word_embeddings": true,
131
+ "transformers_version": "5.10.0",
132
+ "use_cache": false,
133
+ "video_token_id": 258884,
134
+ "vision_config": {
135
+ "_name_or_path": "",
136
+ "architectures": null,
137
+ "attention_bias": false,
138
+ "attention_dropout": 0.0,
139
+ "chunk_size_feed_forward": 0,
140
+ "default_output_length": 280,
141
+ "dtype": "bfloat16",
142
+ "global_head_dim": 72,
143
+ "head_dim": 72,
144
+ "hidden_activation": "gelu_pytorch_tanh",
145
+ "hidden_size": 1152,
146
+ "id2label": {
147
+ "0": "LABEL_0",
148
+ "1": "LABEL_1"
149
+ },
150
+ "initializer_range": 0.02,
151
+ "intermediate_size": 4304,
152
+ "is_encoder_decoder": false,
153
+ "label2id": {
154
+ "LABEL_0": 0,
155
+ "LABEL_1": 1
156
+ },
157
+ "max_position_embeddings": 131072,
158
+ "model_type": "gemma4_vision",
159
+ "num_attention_heads": 16,
160
+ "num_hidden_layers": 27,
161
+ "num_key_value_heads": 16,
162
+ "output_attentions": false,
163
+ "output_hidden_states": false,
164
+ "patch_size": 16,
165
+ "pooling_kernel_size": 3,
166
+ "position_embedding_size": 10240,
167
+ "problem_type": null,
168
+ "return_dict": true,
169
+ "rms_norm_eps": 1e-06,
170
+ "rope_parameters": {
171
+ "rope_theta": 100.0,
172
+ "rope_type": "default"
173
+ },
174
+ "standardize": true,
175
+ "use_clipped_linears": false
176
+ },
177
+ "vision_soft_tokens_per_image": 280
178
+ }
g1_jupyter_notebook/500M/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 1,
6
+ 1
7
+ ],
8
+ "pad_token_id": 0,
9
+ "temperature": 1.0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.10.0"
13
+ }
g1_jupyter_notebook/500M/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b2dfe1aa8ab4edb1328f829c1595c217abcce843636e1fdcb3ac0d034f8aa77
3
+ size 49923154850
g1_jupyter_notebook/500M/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:590d79026305a4a3acf7fb5e4f606aa5fe429e98688d31b3e9d988e64efe0da4
3
+ size 12623183414
g1_jupyter_notebook/500M/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
g1_jupyter_notebook/500M/processor_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 512,
9
+ "fft_overdrive": false,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "right",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
+ "image_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": false,
28
+ "do_rescale": true,
29
+ "do_resize": true,
30
+ "image_mean": [
31
+ 0.0,
32
+ 0.0,
33
+ 0.0
34
+ ],
35
+ "image_processor_type": "Gemma4ImageProcessor",
36
+ "image_seq_length": 280,
37
+ "image_std": [
38
+ 1.0,
39
+ 1.0,
40
+ 1.0
41
+ ],
42
+ "max_soft_tokens": 280,
43
+ "patch_size": 16,
44
+ "pooling_kernel_size": 3,
45
+ "resample": 3,
46
+ "rescale_factor": 0.00392156862745098
47
+ },
48
+ "image_seq_length": 280,
49
+ "processor_class": "Gemma4Processor",
50
+ "video_processor": {
51
+ "do_convert_rgb": true,
52
+ "do_normalize": true,
53
+ "do_rescale": true,
54
+ "do_resize": true,
55
+ "do_sample_frames": true,
56
+ "image_mean": [
57
+ 0.0,
58
+ 0.0,
59
+ 0.0
60
+ ],
61
+ "image_std": [
62
+ 1.0,
63
+ 1.0,
64
+ 1.0
65
+ ],
66
+ "max_soft_tokens": 70,
67
+ "num_frames": 32,
68
+ "patch_size": 16,
69
+ "pooling_kernel_size": 3,
70
+ "resample": 3,
71
+ "rescale_factor": 0.00392156862745098,
72
+ "return_metadata": false,
73
+ "video_processor_type": "Gemma4VideoProcessor"
74
+ }
75
+ }
g1_jupyter_notebook/500M/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12bac982b793c44b03d52a250a9f0d0b666813da566b910c24a6da0695fd11e6
3
+ size 32170070
g1_jupyter_notebook/500M/tokenizer_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_token": "<|audio|>",
3
+ "backend": "tokenizers",
4
+ "boa_token": "<|audio>",
5
+ "boi_token": "<|image>",
6
+ "bos_token": "<bos>",
7
+ "eoa_token": "<audio|>",
8
+ "eoc_token": "<channel|>",
9
+ "eoi_token": "<image|>",
10
+ "eos_token": "<eos>",
11
+ "eot_token": "<turn|>",
12
+ "escape_token": "<|\"|>",
13
+ "etc_token": "<tool_call|>",
14
+ "etd_token": "<tool|>",
15
+ "etr_token": "<tool_response|>",
16
+ "extra_special_tokens": [
17
+ "<|video|>"
18
+ ],
19
+ "image_token": "<|image|>",
20
+ "is_local": true,
21
+ "local_files_only": true,
22
+ "mask_token": "<mask>",
23
+ "model_max_length": 1000000000000000019884624838656,
24
+ "model_specific_special_tokens": {
25
+ "audio_token": "<|audio|>",
26
+ "boa_token": "<|audio>",
27
+ "boi_token": "<|image>",
28
+ "eoa_token": "<audio|>",
29
+ "eoc_token": "<channel|>",
30
+ "eoi_token": "<image|>",
31
+ "eot_token": "<turn|>",
32
+ "escape_token": "<|\"|>",
33
+ "etc_token": "<tool_call|>",
34
+ "etd_token": "<tool|>",
35
+ "etr_token": "<tool_response|>",
36
+ "image_token": "<|image|>",
37
+ "soc_token": "<|channel>",
38
+ "sot_token": "<|turn>",
39
+ "stc_token": "<|tool_call>",
40
+ "std_token": "<|tool>",
41
+ "str_token": "<|tool_response>",
42
+ "think_token": "<|think|>"
43
+ },
44
+ "pad_token": "<pad>",
45
+ "padding_side": "right",
46
+ "processor_class": "Gemma4Processor",
47
+ "soc_token": "<|channel>",
48
+ "sot_token": "<|turn>",
49
+ "split_special_tokens": false,
50
+ "stc_token": "<|tool_call>",
51
+ "std_token": "<|tool>",
52
+ "str_token": "<|tool_response>",
53
+ "think_token": "<|think|>",
54
+ "tokenizer_class": "GemmaTokenizer",
55
+ "unk_token": "<unk>"
56
+ }
g1_jupyter_notebook/500M/train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.4095962551199532,
3
+ "total_flos": 4821608300544000.0,
4
+ "train_loss": 0.039095867361341206,
5
+ "train_runtime": 2660.3321,
6
+ "train_samples_per_second": 25.26,
7
+ "train_steps_per_second": 0.263
8
+ }
g1_jupyter_notebook/500M/trainer_log.jsonl ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {"current_steps": 640, "total_steps": 700, "loss": 0.44627851247787476, "lr": 1.9892306274149575e-05, "epoch": 0.37448800468110005, "percentage": 91.43, "elapsed_time": "0:02:53", "remaining_time": "0:00:16"}
2
+ {"current_steps": 650, "total_steps": 700, "loss": 0.42882070541381834, "lr": 1.8041914938232418e-05, "epoch": 0.38033937975424226, "percentage": 92.86, "elapsed_time": "0:09:37", "remaining_time": "0:00:44"}
3
+ {"current_steps": 660, "total_steps": 700, "loss": 0.434341287612915, "lr": 1.430417317963131e-05, "epoch": 0.3861907548273844, "percentage": 94.29, "elapsed_time": "0:16:21", "remaining_time": "0:00:59"}
4
+ {"current_steps": 670, "total_steps": 700, "loss": 0.4225472450256348, "lr": 9.561776640442192e-06, "epoch": 0.3920421299005266, "percentage": 95.71, "elapsed_time": "0:23:04", "remaining_time": "0:01:01"}
5
+ {"current_steps": 680, "total_steps": 700, "loss": 0.4186511993408203, "lr": 4.934677695277445e-06, "epoch": 0.3978935049736688, "percentage": 97.14, "elapsed_time": "0:29:47", "remaining_time": "0:00:52"}
6
+ {"current_steps": 690, "total_steps": 700, "loss": 0.4262665271759033, "lr": 1.5156003304972938e-06, "epoch": 0.403744880046811, "percentage": 98.57, "elapsed_time": "0:36:31", "remaining_time": "0:00:31"}
7
+ {"current_steps": 700, "total_steps": 700, "loss": 0.42757234573364256, "lr": 1.1198521075853458e-07, "epoch": 0.4095962551199532, "percentage": 100.0, "elapsed_time": "0:43:16", "remaining_time": "0:00:00"}
8
+ {"current_steps": 700, "total_steps": 700, "epoch": 0.4095962551199532, "percentage": 100.0, "elapsed_time": "0:44:20", "remaining_time": "0:00:00"}
g1_jupyter_notebook/500M/trainer_state.json ADDED
@@ -0,0 +1,533 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.4095962551199532,
6
+ "eval_steps": 500,
7
+ "global_step": 700,
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.005851375073142188,
14
+ "grad_norm": 722.3047506884299,
15
+ "learning_rate": 8.240909090909092e-06,
16
+ "loss": 0.815431022644043,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.011702750146284377,
21
+ "grad_norm": 145.4011608176478,
22
+ "learning_rate": 1.728636363636364e-05,
23
+ "loss": 0.4873072624206543,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.017554125219426564,
28
+ "grad_norm": 34.7425918176051,
29
+ "learning_rate": 2e-05,
30
+ "loss": 0.47413363456726076,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.023405500292568753,
35
+ "grad_norm": 627.6113988831149,
36
+ "learning_rate": 2e-05,
37
+ "loss": 0.5241857051849366,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.029256875365710942,
42
+ "grad_norm": 9.333282061786075,
43
+ "learning_rate": 2e-05,
44
+ "loss": 0.5261882781982422,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.03510825043885313,
49
+ "grad_norm": 10.258378288452999,
50
+ "learning_rate": 2e-05,
51
+ "loss": 0.5006448745727539,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.04095962551199532,
56
+ "grad_norm": 13.916482807305034,
57
+ "learning_rate": 2e-05,
58
+ "loss": 0.48270502090454104,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.046811000585137506,
63
+ "grad_norm": 7.448753281691261,
64
+ "learning_rate": 2e-05,
65
+ "loss": 0.5237825393676758,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.0526623756582797,
70
+ "grad_norm": 13.959049034574702,
71
+ "learning_rate": 2e-05,
72
+ "loss": 0.4954239845275879,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.058513750731421885,
77
+ "grad_norm": 6.419507595225647,
78
+ "learning_rate": 2e-05,
79
+ "loss": 0.47747349739074707,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.06436512580456408,
84
+ "grad_norm": 10.333527235184453,
85
+ "learning_rate": 2e-05,
86
+ "loss": 0.46398019790649414,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.07021650087770626,
91
+ "grad_norm": 4.7743767265565715,
92
+ "learning_rate": 2e-05,
93
+ "loss": 0.45823111534118655,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.07606787595084845,
98
+ "grad_norm": 35.948840970822566,
99
+ "learning_rate": 2e-05,
100
+ "loss": 0.47070822715759275,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.08191925102399064,
105
+ "grad_norm": 14.441710239001834,
106
+ "learning_rate": 2e-05,
107
+ "loss": 0.4741715908050537,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.08777062609713282,
112
+ "grad_norm": 19.748869955541718,
113
+ "learning_rate": 2e-05,
114
+ "loss": 0.4672824859619141,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.09362200117027501,
119
+ "grad_norm": 5.376095253416546,
120
+ "learning_rate": 2e-05,
121
+ "loss": 0.4619298458099365,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.0994733762434172,
126
+ "grad_norm": 3.5447746494342622,
127
+ "learning_rate": 2e-05,
128
+ "loss": 0.4703338146209717,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.1053247513165594,
133
+ "grad_norm": 22.237367801418927,
134
+ "learning_rate": 2e-05,
135
+ "loss": 0.4801036834716797,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.11117612638970158,
140
+ "grad_norm": 10.469269003466454,
141
+ "learning_rate": 2e-05,
142
+ "loss": 0.4776778221130371,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.11702750146284377,
147
+ "grad_norm": 69.00024674808894,
148
+ "learning_rate": 2e-05,
149
+ "loss": 0.4732304573059082,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.12287887653598596,
154
+ "grad_norm": 4.008167284876325,
155
+ "learning_rate": 2e-05,
156
+ "loss": 0.4633647441864014,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.12873025160912815,
161
+ "grad_norm": 4.12037159490575,
162
+ "learning_rate": 2e-05,
163
+ "loss": 0.4576042652130127,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.13458162668227033,
168
+ "grad_norm": 10.02371679111215,
169
+ "learning_rate": 2e-05,
170
+ "loss": 0.4506582260131836,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.1404330017554125,
175
+ "grad_norm": 145.26436740003052,
176
+ "learning_rate": 2e-05,
177
+ "loss": 0.4545659065246582,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.14628437682855472,
182
+ "grad_norm": 1.1636166760298137,
183
+ "learning_rate": 2e-05,
184
+ "loss": 0.46163148880004884,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.1521357519016969,
189
+ "grad_norm": 2.4689908414338637,
190
+ "learning_rate": 2e-05,
191
+ "loss": 0.4646605491638184,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.15798712697483908,
196
+ "grad_norm": 7.311694985604752,
197
+ "learning_rate": 2e-05,
198
+ "loss": 0.4644497871398926,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.16383850204798128,
203
+ "grad_norm": 120.76111223511168,
204
+ "learning_rate": 2e-05,
205
+ "loss": 0.45812482833862306,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.16968987712112346,
210
+ "grad_norm": 7.432860843908352,
211
+ "learning_rate": 2e-05,
212
+ "loss": 0.4690185546875,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.17554125219426564,
217
+ "grad_norm": 3.5063397185454206,
218
+ "learning_rate": 2e-05,
219
+ "loss": 0.47104778289794924,
220
+ "step": 300
221
+ },
222
+ {
223
+ "epoch": 0.18139262726740785,
224
+ "grad_norm": 1.7161368671120483,
225
+ "learning_rate": 2e-05,
226
+ "loss": 0.47832560539245605,
227
+ "step": 310
228
+ },
229
+ {
230
+ "epoch": 0.18724400234055003,
231
+ "grad_norm": 2.7995694136387734,
232
+ "learning_rate": 2e-05,
233
+ "loss": 0.4722126007080078,
234
+ "step": 320
235
+ },
236
+ {
237
+ "epoch": 0.1930953774136922,
238
+ "grad_norm": 10.80819931602094,
239
+ "learning_rate": 2e-05,
240
+ "loss": 0.46016683578491213,
241
+ "step": 330
242
+ },
243
+ {
244
+ "epoch": 0.1989467524868344,
245
+ "grad_norm": 1.364105506206788,
246
+ "learning_rate": 2e-05,
247
+ "loss": 0.4547887802124023,
248
+ "step": 340
249
+ },
250
+ {
251
+ "epoch": 0.2047981275599766,
252
+ "grad_norm": 1.657602624234001,
253
+ "learning_rate": 2e-05,
254
+ "loss": 0.4447385311126709,
255
+ "step": 350
256
+ },
257
+ {
258
+ "epoch": 0.2106495026331188,
259
+ "grad_norm": 6.984782890612126,
260
+ "learning_rate": 2e-05,
261
+ "loss": 0.4384812831878662,
262
+ "step": 360
263
+ },
264
+ {
265
+ "epoch": 0.21650087770626097,
266
+ "grad_norm": 5.233046035168877,
267
+ "learning_rate": 2e-05,
268
+ "loss": 0.44702463150024413,
269
+ "step": 370
270
+ },
271
+ {
272
+ "epoch": 0.22235225277940315,
273
+ "grad_norm": 1.209529576166064,
274
+ "learning_rate": 2e-05,
275
+ "loss": 0.44935851626926,
276
+ "step": 380
277
+ },
278
+ {
279
+ "epoch": 0.22820362785254536,
280
+ "grad_norm": 10.9279502257516,
281
+ "learning_rate": 2e-05,
282
+ "loss": 0.44748721122741697,
283
+ "step": 390
284
+ },
285
+ {
286
+ "epoch": 0.23405500292568754,
287
+ "grad_norm": 7.470557475551403,
288
+ "learning_rate": 2e-05,
289
+ "loss": 0.45525145530700684,
290
+ "step": 400
291
+ },
292
+ {
293
+ "epoch": 0.23990637799882972,
294
+ "grad_norm": 2.059197815167869,
295
+ "learning_rate": 2e-05,
296
+ "loss": 0.4523183822631836,
297
+ "step": 410
298
+ },
299
+ {
300
+ "epoch": 0.24575775307197192,
301
+ "grad_norm": 8.618618633793242,
302
+ "learning_rate": 2e-05,
303
+ "loss": 0.45379066467285156,
304
+ "step": 420
305
+ },
306
+ {
307
+ "epoch": 0.2516091281451141,
308
+ "grad_norm": 2.805939217260095,
309
+ "learning_rate": 2e-05,
310
+ "loss": 0.4506677786509196,
311
+ "step": 430
312
+ },
313
+ {
314
+ "epoch": 0.2574605032182563,
315
+ "grad_norm": 2.9649350106279253,
316
+ "learning_rate": 2e-05,
317
+ "loss": 0.45397562980651857,
318
+ "step": 440
319
+ },
320
+ {
321
+ "epoch": 0.26331187829139846,
322
+ "grad_norm": 1.2882383449845194,
323
+ "learning_rate": 2e-05,
324
+ "loss": 0.4674218654632568,
325
+ "step": 450
326
+ },
327
+ {
328
+ "epoch": 0.26916325336454067,
329
+ "grad_norm": 2.498448492094687,
330
+ "learning_rate": 2e-05,
331
+ "loss": 0.4801971912384033,
332
+ "step": 460
333
+ },
334
+ {
335
+ "epoch": 0.2750146284376829,
336
+ "grad_norm": 4.1544402071454645,
337
+ "learning_rate": 2e-05,
338
+ "loss": 0.45485677719116213,
339
+ "step": 470
340
+ },
341
+ {
342
+ "epoch": 0.280866003510825,
343
+ "grad_norm": 12.647271506037224,
344
+ "learning_rate": 2e-05,
345
+ "loss": 0.45079994201660156,
346
+ "step": 480
347
+ },
348
+ {
349
+ "epoch": 0.28671737858396723,
350
+ "grad_norm": 4.910636891464567,
351
+ "learning_rate": 2e-05,
352
+ "loss": 0.44969873428344725,
353
+ "step": 490
354
+ },
355
+ {
356
+ "epoch": 0.29256875365710944,
357
+ "grad_norm": 1.8609845979795288,
358
+ "learning_rate": 2e-05,
359
+ "loss": 0.43883328437805175,
360
+ "step": 500
361
+ },
362
+ {
363
+ "epoch": 0.2984201287302516,
364
+ "grad_norm": 1.7542751511057226,
365
+ "learning_rate": 2e-05,
366
+ "loss": 0.4447303295135498,
367
+ "step": 510
368
+ },
369
+ {
370
+ "epoch": 0.3042715038033938,
371
+ "grad_norm": 1.0135283772570185,
372
+ "learning_rate": 2e-05,
373
+ "loss": 0.4480525016784668,
374
+ "step": 520
375
+ },
376
+ {
377
+ "epoch": 0.310122878876536,
378
+ "grad_norm": 1.0019226878150138,
379
+ "learning_rate": 2e-05,
380
+ "loss": 0.4496911525726318,
381
+ "step": 530
382
+ },
383
+ {
384
+ "epoch": 0.31597425394967815,
385
+ "grad_norm": 0.8128881881211669,
386
+ "learning_rate": 2e-05,
387
+ "loss": 0.4464526653289795,
388
+ "step": 540
389
+ },
390
+ {
391
+ "epoch": 0.32182562902282036,
392
+ "grad_norm": 2.2177233161225653,
393
+ "learning_rate": 2e-05,
394
+ "loss": 0.43948893547058104,
395
+ "step": 550
396
+ },
397
+ {
398
+ "epoch": 0.32767700409596257,
399
+ "grad_norm": 10.900543579060534,
400
+ "learning_rate": 2e-05,
401
+ "loss": 0.44472250938415525,
402
+ "step": 560
403
+ },
404
+ {
405
+ "epoch": 0.3335283791691047,
406
+ "grad_norm": 2.1356515208509332,
407
+ "learning_rate": 2e-05,
408
+ "loss": 0.43558540344238283,
409
+ "step": 570
410
+ },
411
+ {
412
+ "epoch": 0.3393797542422469,
413
+ "grad_norm": 1.3374263807054463,
414
+ "learning_rate": 2e-05,
415
+ "loss": 0.44030489921569826,
416
+ "step": 580
417
+ },
418
+ {
419
+ "epoch": 0.34523112931538913,
420
+ "grad_norm": 25.135467104722675,
421
+ "learning_rate": 2e-05,
422
+ "loss": 0.4326209545135498,
423
+ "step": 590
424
+ },
425
+ {
426
+ "epoch": 0.3510825043885313,
427
+ "grad_norm": 2.1580634731529167,
428
+ "learning_rate": 2e-05,
429
+ "loss": 0.4280872344970703,
430
+ "step": 600
431
+ },
432
+ {
433
+ "epoch": 0.3569338794616735,
434
+ "grad_norm": 4.127553728499885,
435
+ "learning_rate": 2e-05,
436
+ "loss": 0.42243247032165526,
437
+ "step": 610
438
+ },
439
+ {
440
+ "epoch": 0.3627852545348157,
441
+ "grad_norm": 14.048209030626133,
442
+ "learning_rate": 2e-05,
443
+ "loss": 0.4255032539367676,
444
+ "step": 620
445
+ },
446
+ {
447
+ "epoch": 0.36863662960795784,
448
+ "grad_norm": 4.178781966521355,
449
+ "learning_rate": 2e-05,
450
+ "loss": 0.43591742515563964,
451
+ "step": 630
452
+ },
453
+ {
454
+ "epoch": 0.37448800468110005,
455
+ "grad_norm": 1.994470465668841,
456
+ "learning_rate": 1.9892306274149575e-05,
457
+ "loss": 0.44627851247787476,
458
+ "step": 640
459
+ },
460
+ {
461
+ "epoch": 0.38033937975424226,
462
+ "grad_norm": 13.282126521384157,
463
+ "learning_rate": 1.8041914938232418e-05,
464
+ "loss": 0.42882070541381834,
465
+ "step": 650
466
+ },
467
+ {
468
+ "epoch": 0.3861907548273844,
469
+ "grad_norm": 2.3650714056808746,
470
+ "learning_rate": 1.430417317963131e-05,
471
+ "loss": 0.434341287612915,
472
+ "step": 660
473
+ },
474
+ {
475
+ "epoch": 0.3920421299005266,
476
+ "grad_norm": 3.8698654531415233,
477
+ "learning_rate": 9.561776640442192e-06,
478
+ "loss": 0.4225472450256348,
479
+ "step": 670
480
+ },
481
+ {
482
+ "epoch": 0.3978935049736688,
483
+ "grad_norm": 4.245759234905024,
484
+ "learning_rate": 4.934677695277445e-06,
485
+ "loss": 0.4186511993408203,
486
+ "step": 680
487
+ },
488
+ {
489
+ "epoch": 0.403744880046811,
490
+ "grad_norm": 4.249533551125241,
491
+ "learning_rate": 1.5156003304972938e-06,
492
+ "loss": 0.4262665271759033,
493
+ "step": 690
494
+ },
495
+ {
496
+ "epoch": 0.4095962551199532,
497
+ "grad_norm": 6.489441960690249,
498
+ "learning_rate": 1.1198521075853458e-07,
499
+ "loss": 0.42757234573364256,
500
+ "step": 700
501
+ },
502
+ {
503
+ "epoch": 0.4095962551199532,
504
+ "step": 700,
505
+ "total_flos": 4821608300544000.0,
506
+ "train_loss": 0.039095867361341206,
507
+ "train_runtime": 2660.3321,
508
+ "train_samples_per_second": 25.26,
509
+ "train_steps_per_second": 0.263
510
+ }
511
+ ],
512
+ "logging_steps": 10,
513
+ "max_steps": 700,
514
+ "num_input_tokens_seen": 0,
515
+ "num_train_epochs": 1,
516
+ "save_steps": 1000000,
517
+ "stateful_callbacks": {
518
+ "TrainerControl": {
519
+ "args": {
520
+ "should_epoch_stop": false,
521
+ "should_evaluate": false,
522
+ "should_log": false,
523
+ "should_save": true,
524
+ "should_training_stop": true
525
+ },
526
+ "attributes": {}
527
+ }
528
+ },
529
+ "total_flos": 4821608300544000.0,
530
+ "train_batch_size": 1,
531
+ "trial_name": null,
532
+ "trial_params": null
533
+ }
g1_jupyter_notebook/500M/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91b659b8b4abe79b3d34743a66e8c4a52dfa23907e83a69df2e85878a17cc4d0
3
+ size 8209
g1_jupyter_notebook/500M/training_loss.png ADDED