kth8 commited on
Commit
bc34009
·
verified ·
1 Parent(s): 75ce68d

Upload folder using huggingface_hub

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,142 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ language:
4
+ - en
5
+ base_model: unsloth/gemma-3-1b-it
6
+ datasets:
7
+ - kth8/title-generation-25000x
8
+ pipeline_tag: text-generation
9
+ library_name: transformers
10
+ tags:
11
+ - sft
12
+ - trl
13
+ - unsloth
14
+ - gemma
15
+ - gemma3
16
+ - gemma3_text
17
+ ---
18
+ ![logo](https://storage.googleapis.com/gweb-developer-goog-blog-assets/images/gemma-3_2.original.png)
19
+ A supervised fine-tune of [unsloth/gemma-3-1b-it](https://huggingface.co/unsloth/gemma-3-1b-it) on the [kth8/title-generation-25000x](https://huggingface.co/datasets/kth8/title-generation-25000x) dataset.
20
+ Trained with the exact system prompt OpenCode's [title agent uses](https://raw.githubusercontent.com/anomalyco/opencode/refs/heads/dev/packages/opencode/src/agent/prompt/title.txt).
21
+
22
+ ## Usage example
23
+
24
+ **System prompt**
25
+ ```
26
+ You are a title generator. You output ONLY a thread title. Nothing else.
27
+
28
+ <task>
29
+ Generate a brief title that would help the user find this conversation later.
30
+
31
+ Follow all rules in <rules>
32
+ Use the <examples> so you know what a good title looks like.
33
+ Your output must be:
34
+ - A single line
35
+ - ≤50 characters
36
+ - No explanations
37
+ </task>
38
+
39
+ <rules>
40
+ - you MUST use the same language as the user message you are summarizing
41
+ - Title must be grammatically correct and read naturally - no word salad
42
+ - Never include tool names in the title (e.g. "read tool", "bash tool", "edit tool")
43
+ - Focus on the main topic or question the user needs to retrieve
44
+ - Vary your phrasing - avoid repetitive patterns like always starting with "Analyzing"
45
+ - When a file is mentioned, focus on WHAT the user wants to do WITH the file, not just that they shared it
46
+ - Keep exact: technical terms, numbers, filenames, HTTP codes
47
+ - Remove: the, this, my, a, an
48
+ - Never assume tech stack
49
+ - Never use tools
50
+ - NEVER respond to questions, just generate a title for the conversation
51
+ - The title should NEVER include "summarizing" or "generating" when generating a title
52
+ - DO NOT SAY YOU CANNOT GENERATE A TITLE OR COMPLAIN ABOUT THE INPUT
53
+ - Always output something meaningful, even if the input is minimal.
54
+ - If the user message is short or conversational (e.g. "hello", "lol", "what's up", "hey"):
55
+ → create a title that reflects the user's tone or intent (such as Greeting, Quick check-in, Light chat, Intro message, etc.)
56
+ </rules>
57
+
58
+ <examples>
59
+ "debug 500 errors in production" → Debugging production 500 errors
60
+ "refactor user service" → Refactoring user service
61
+ "why is app.js failing" → app.js failure investigation
62
+ "implement rate limiting" → Rate limiting implementation
63
+ "how do I connect postgres to my API" → Postgres API connection
64
+ "best practices for React hooks" → React hooks best practices
65
+ "@src/auth.ts can you add refresh token support" → Auth refresh token support
66
+ "@utils/parser.ts this is broken" → Parser bug fix
67
+ "look at @config.json" → Config review
68
+ "@App.tsx add dark mode toggle" → Dark mode toggle in App
69
+ </examples>
70
+ ```
71
+ **User prompt**
72
+ ```
73
+ If there were 200 students who passed an English course three years ago, and each subsequent year until the current one that number increased by 50% of the previous year's number, how many students will pass the course this year?
74
+ ```
75
+ **Assistant response**
76
+ ```
77
+ Student course passing growth calculation
78
+ ```
79
+ ## Model Details
80
+ - Base Model: `unsloth/gemma-3-1b-it`
81
+ - Parameter Count: 999,885,952
82
+ - Precision: torch.bfloat16
83
+
84
+ ## Training Settings
85
+
86
+ ### PEFT
87
+ - Rank: 32
88
+ - LoRA alpha: 64
89
+ - Modules: q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj
90
+ - Gradient checkpointing: unsloth
91
+
92
+ ### SFT
93
+ - Epoch: 1
94
+ - Batch size: 8
95
+ - Gradient Accumulation steps: 2
96
+ - Learning rate: 0.0002
97
+ - Optimizer: adamw_torch_fused
98
+ - Learning rate scheduler: cosine
99
+ - Warmup steps: 100
100
+ - Weight decay: 0.01
101
+
102
+ ## Training stats
103
+ - Date: 2026-06-01T08:29:30.781636
104
+ - GPU: NVIDIA A100-SXM4-40GB
105
+ - Peak VRAM usage: 13.873 GB
106
+ - Global step: 1588
107
+ - Training runtime (seconds): 2468.3919
108
+ - Best validation loss: 1.019551396369934
109
+
110
+ | Step | Training Loss | Validation Loss |
111
+ |------|---------------|-----------------|
112
+ | 79 | 1.295400 | 1.364987 |
113
+ | 158 | 1.269300 | 1.269067 |
114
+ | 237 | 1.199000 | 1.264947 |
115
+ | 316 | 1.249800 | 1.216193 |
116
+ | 395 | 1.342900 | 1.188592 |
117
+ | 474 | 1.212000 | 1.192388 |
118
+ | 553 | 1.194300 | 1.155604 |
119
+ | 632 | 1.174100 | 1.136098 |
120
+ | 711 | 1.088800 | 1.133109 |
121
+ | 790 | 1.148900 | 1.151859 |
122
+ | 869 | 1.070300 | 1.112693 |
123
+ | 948 | 1.005500 | 1.072988 |
124
+ | 1027 | 1.035400 | 1.059929 |
125
+ | 1106 | 1.075200 | 1.054300 |
126
+ | 1185 | 1.078300 | 1.046455 |
127
+ | 1264 | 1.063400 | 1.032251 |
128
+ | 1343 | 0.980000 | 1.026014 |
129
+ | 1422 | 1.040300 | 1.019870 |
130
+ | 1501 | 0.987600 | 1.019551 |
131
+ | 1580 | 0.988000 | 1.019595 |
132
+
133
+ ## Framework versions
134
+ - Unsloth: 2026.5.9
135
+ - TRL: 0.22.2
136
+ - Transformers: 4.56.2
137
+ - Pytorch: 2.11.0+cu128
138
+ - Datasets: 4.8.5
139
+ - Tokenizers: 0.22.2
140
+
141
+ ## License
142
+ This model is released under the Gemma license. See the [Gemma Terms of Use](https://ai.google.dev/gemma/terms) and [Prohibited Use Policy](https://policies.google.com/terms/generative-ai/use-policy) regarding the use of Gemma-generated content.
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
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 -%}
config.json ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3ForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "cache_implementation": "hybrid",
11
+ "dtype": "bfloat16",
12
+ "eos_token_id": 106,
13
+ "final_logit_softcapping": null,
14
+ "head_dim": 256,
15
+ "hidden_activation": "gelu_pytorch_tanh",
16
+ "hidden_size": 1152,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 6912,
19
+ "layer_types": [
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "sliding_attention",
25
+ "full_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention",
29
+ "sliding_attention",
30
+ "sliding_attention",
31
+ "full_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "sliding_attention",
37
+ "full_attention",
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
+ ],
47
+ "max_position_embeddings": 32768,
48
+ "model_type": "gemma3_text",
49
+ "num_attention_heads": 4,
50
+ "num_hidden_layers": 26,
51
+ "num_key_value_heads": 1,
52
+ "pad_token_id": 0,
53
+ "query_pre_attn_scalar": 256,
54
+ "rms_norm_eps": 1e-06,
55
+ "rope_local_base_freq": 10000,
56
+ "rope_scaling": null,
57
+ "rope_theta": 1000000,
58
+ "sliding_window": 512,
59
+ "transformers_version": "4.56.2",
60
+ "unsloth_fixed": true,
61
+ "use_cache": true,
62
+ "vocab_size": 262144
63
+ }
generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "max_length": 32768,
10
+ "pad_token_id": 0,
11
+ "top_k": 64,
12
+ "top_p": 0.95,
13
+ "transformers_version": "4.56.2"
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c234d92585f5bf0ed57ed1d171010da2fc92ca7c78b054d8468273b1ac9cf8c8
3
+ size 1999811208
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<end_of_turn>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
3
+ size 33384568
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff
 
train/log.json ADDED
@@ -0,0 +1,1277 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "loss": 4.261,
4
+ "grad_norm": 28.33258819580078,
5
+ "learning_rate": 1.8e-05,
6
+ "epoch": 0.006297229219143577,
7
+ "step": 10
8
+ },
9
+ {
10
+ "loss": 2.013,
11
+ "grad_norm": 8.625818252563477,
12
+ "learning_rate": 3.8e-05,
13
+ "epoch": 0.012594458438287154,
14
+ "step": 20
15
+ },
16
+ {
17
+ "loss": 1.5624,
18
+ "grad_norm": 7.547111511230469,
19
+ "learning_rate": 5.8e-05,
20
+ "epoch": 0.018891687657430732,
21
+ "step": 30
22
+ },
23
+ {
24
+ "loss": 1.4473,
25
+ "grad_norm": 9.756204605102539,
26
+ "learning_rate": 7.800000000000001e-05,
27
+ "epoch": 0.02518891687657431,
28
+ "step": 40
29
+ },
30
+ {
31
+ "loss": 1.3737,
32
+ "grad_norm": 5.382819652557373,
33
+ "learning_rate": 9.8e-05,
34
+ "epoch": 0.031486146095717885,
35
+ "step": 50
36
+ },
37
+ {
38
+ "loss": 1.4106,
39
+ "grad_norm": 9.088937759399414,
40
+ "learning_rate": 0.000118,
41
+ "epoch": 0.037783375314861464,
42
+ "step": 60
43
+ },
44
+ {
45
+ "loss": 1.2954,
46
+ "grad_norm": 6.663686275482178,
47
+ "learning_rate": 0.000138,
48
+ "epoch": 0.04408060453400504,
49
+ "step": 70
50
+ },
51
+ {
52
+ "eval_loss": 1.3649871349334717,
53
+ "eval_runtime": 37.2236,
54
+ "eval_samples_per_second": 13.943,
55
+ "eval_steps_per_second": 3.492,
56
+ "epoch": 0.04974811083123426,
57
+ "step": 79
58
+ },
59
+ {
60
+ "loss": 1.4254,
61
+ "grad_norm": 6.387028694152832,
62
+ "learning_rate": 0.00015800000000000002,
63
+ "epoch": 0.05037783375314862,
64
+ "step": 80
65
+ },
66
+ {
67
+ "loss": 1.2516,
68
+ "grad_norm": 4.3521552085876465,
69
+ "learning_rate": 0.00017800000000000002,
70
+ "epoch": 0.05667506297229219,
71
+ "step": 90
72
+ },
73
+ {
74
+ "loss": 1.2928,
75
+ "grad_norm": 5.566264629364014,
76
+ "learning_rate": 0.00019800000000000002,
77
+ "epoch": 0.06297229219143577,
78
+ "step": 100
79
+ },
80
+ {
81
+ "loss": 1.3505,
82
+ "grad_norm": 4.266009330749512,
83
+ "learning_rate": 0.0001999819475629623,
84
+ "epoch": 0.06926952141057935,
85
+ "step": 110
86
+ },
87
+ {
88
+ "loss": 1.3153,
89
+ "grad_norm": 6.01008939743042,
90
+ "learning_rate": 0.00019991955244474347,
91
+ "epoch": 0.07556675062972293,
92
+ "step": 120
93
+ },
94
+ {
95
+ "loss": 1.307,
96
+ "grad_norm": 3.635507583618164,
97
+ "learning_rate": 0.0001998126195808803,
98
+ "epoch": 0.0818639798488665,
99
+ "step": 130
100
+ },
101
+ {
102
+ "loss": 1.2197,
103
+ "grad_norm": 5.475412845611572,
104
+ "learning_rate": 0.00019966119663520412,
105
+ "epoch": 0.08816120906801007,
106
+ "step": 140
107
+ },
108
+ {
109
+ "loss": 1.2693,
110
+ "grad_norm": 4.274078845977783,
111
+ "learning_rate": 0.00019946535110237904,
112
+ "epoch": 0.09445843828715365,
113
+ "step": 150
114
+ },
115
+ {
116
+ "eval_loss": 1.269067406654358,
117
+ "eval_runtime": 18.4869,
118
+ "eval_samples_per_second": 28.074,
119
+ "eval_steps_per_second": 7.032,
120
+ "epoch": 0.09949622166246852,
121
+ "step": 158
122
+ },
123
+ {
124
+ "loss": 1.2633,
125
+ "grad_norm": 5.927584648132324,
126
+ "learning_rate": 0.00019922517027781746,
127
+ "epoch": 0.10075566750629723,
128
+ "step": 160
129
+ },
130
+ {
131
+ "loss": 1.2406,
132
+ "grad_norm": 4.917751789093018,
133
+ "learning_rate": 0.000198940761218769,
134
+ "epoch": 0.1070528967254408,
135
+ "step": 170
136
+ },
137
+ {
138
+ "loss": 1.2731,
139
+ "grad_norm": 4.493299961090088,
140
+ "learning_rate": 0.0001986122506966014,
141
+ "epoch": 0.11335012594458438,
142
+ "step": 180
143
+ },
144
+ {
145
+ "loss": 1.2295,
146
+ "grad_norm": 5.21290397644043,
147
+ "learning_rate": 0.000198239785140294,
148
+ "epoch": 0.11964735516372796,
149
+ "step": 190
150
+ },
151
+ {
152
+ "loss": 1.1721,
153
+ "grad_norm": 4.943316459655762,
154
+ "learning_rate": 0.000197823530571169,
155
+ "epoch": 0.12594458438287154,
156
+ "step": 200
157
+ },
158
+ {
159
+ "loss": 1.2901,
160
+ "grad_norm": 4.010244846343994,
161
+ "learning_rate": 0.00019736367252888974,
162
+ "epoch": 0.13224181360201512,
163
+ "step": 210
164
+ },
165
+ {
166
+ "loss": 1.2298,
167
+ "grad_norm": 4.7228779792785645,
168
+ "learning_rate": 0.000196860415988759,
169
+ "epoch": 0.1385390428211587,
170
+ "step": 220
171
+ },
172
+ {
173
+ "loss": 1.199,
174
+ "grad_norm": 4.324512004852295,
175
+ "learning_rate": 0.00019631398527035422,
176
+ "epoch": 0.14483627204030228,
177
+ "step": 230
178
+ },
179
+ {
180
+ "eval_loss": 1.2649470567703247,
181
+ "eval_runtime": 18.6336,
182
+ "eval_samples_per_second": 27.853,
183
+ "eval_steps_per_second": 6.977,
184
+ "epoch": 0.14924433249370278,
185
+ "step": 237
186
+ },
187
+ {
188
+ "loss": 1.2394,
189
+ "grad_norm": 4.480823993682861,
190
+ "learning_rate": 0.00019572462393754005,
191
+ "epoch": 0.15113350125944586,
192
+ "step": 240
193
+ },
194
+ {
195
+ "loss": 1.1254,
196
+ "grad_norm": 5.136316776275635,
197
+ "learning_rate": 0.00019509259468990323,
198
+ "epoch": 0.1574307304785894,
199
+ "step": 250
200
+ },
201
+ {
202
+ "loss": 1.2375,
203
+ "grad_norm": 4.789736747741699,
204
+ "learning_rate": 0.00019441817924565786,
205
+ "epoch": 0.163727959697733,
206
+ "step": 260
207
+ },
208
+ {
209
+ "loss": 1.1322,
210
+ "grad_norm": 3.7004570960998535,
211
+ "learning_rate": 0.00019370167821607353,
212
+ "epoch": 0.17002518891687657,
213
+ "step": 270
214
+ },
215
+ {
216
+ "loss": 1.2878,
217
+ "grad_norm": 4.034934997558594,
218
+ "learning_rate": 0.00019294341097148194,
219
+ "epoch": 0.17632241813602015,
220
+ "step": 280
221
+ },
222
+ {
223
+ "loss": 1.3235,
224
+ "grad_norm": 4.752628803253174,
225
+ "learning_rate": 0.0001921437154989221,
226
+ "epoch": 0.18261964735516373,
227
+ "step": 290
228
+ },
229
+ {
230
+ "loss": 1.3083,
231
+ "grad_norm": 5.0100483894348145,
232
+ "learning_rate": 0.0001913029482514873,
233
+ "epoch": 0.1889168765743073,
234
+ "step": 300
235
+ },
236
+ {
237
+ "loss": 1.2498,
238
+ "grad_norm": 4.2906036376953125,
239
+ "learning_rate": 0.00019042148398944116,
240
+ "epoch": 0.1952141057934509,
241
+ "step": 310
242
+ },
243
+ {
244
+ "eval_loss": 1.216193437576294,
245
+ "eval_runtime": 18.6046,
246
+ "eval_samples_per_second": 27.896,
247
+ "eval_steps_per_second": 6.988,
248
+ "epoch": 0.19899244332493704,
249
+ "step": 316
250
+ },
251
+ {
252
+ "loss": 1.1354,
253
+ "grad_norm": 4.017844200134277,
254
+ "learning_rate": 0.0001894997156131734,
255
+ "epoch": 0.20151133501259447,
256
+ "step": 320
257
+ },
258
+ {
259
+ "loss": 1.2565,
260
+ "grad_norm": 4.2938055992126465,
261
+ "learning_rate": 0.00018853805398807004,
262
+ "epoch": 0.20780856423173805,
263
+ "step": 330
264
+ },
265
+ {
266
+ "loss": 1.1638,
267
+ "grad_norm": 5.103464126586914,
268
+ "learning_rate": 0.00018753692776137584,
269
+ "epoch": 0.2141057934508816,
270
+ "step": 340
271
+ },
272
+ {
273
+ "loss": 1.2232,
274
+ "grad_norm": 4.511836528778076,
275
+ "learning_rate": 0.00018649678317113084,
276
+ "epoch": 0.22040302267002518,
277
+ "step": 350
278
+ },
279
+ {
280
+ "loss": 1.2183,
281
+ "grad_norm": 4.007864475250244,
282
+ "learning_rate": 0.0001854180838472659,
283
+ "epoch": 0.22670025188916876,
284
+ "step": 360
285
+ },
286
+ {
287
+ "loss": 1.2473,
288
+ "grad_norm": 5.685576438903809,
289
+ "learning_rate": 0.00018430131060494616,
290
+ "epoch": 0.23299748110831234,
291
+ "step": 370
292
+ },
293
+ {
294
+ "loss": 1.2598,
295
+ "grad_norm": 5.161244869232178,
296
+ "learning_rate": 0.00018314696123025454,
297
+ "epoch": 0.23929471032745592,
298
+ "step": 380
299
+ },
300
+ {
301
+ "loss": 1.3429,
302
+ "grad_norm": 4.944404125213623,
303
+ "learning_rate": 0.00018195555025831022,
304
+ "epoch": 0.2455919395465995,
305
+ "step": 390
306
+ },
307
+ {
308
+ "eval_loss": 1.1885915994644165,
309
+ "eval_runtime": 18.7194,
310
+ "eval_samples_per_second": 27.725,
311
+ "eval_steps_per_second": 6.945,
312
+ "epoch": 0.24874055415617127,
313
+ "step": 395
314
+ },
315
+ {
316
+ "loss": 1.1816,
317
+ "grad_norm": 3.236201047897339,
318
+ "learning_rate": 0.00018072760874392215,
319
+ "epoch": 0.2518891687657431,
320
+ "step": 400
321
+ },
322
+ {
323
+ "loss": 1.1671,
324
+ "grad_norm": 3.8253695964813232,
325
+ "learning_rate": 0.00017946368402487845,
326
+ "epoch": 0.25818639798488663,
327
+ "step": 410
328
+ },
329
+ {
330
+ "loss": 1.1089,
331
+ "grad_norm": 4.663104057312012,
332
+ "learning_rate": 0.00017816433947797858,
333
+ "epoch": 0.26448362720403024,
334
+ "step": 420
335
+ },
336
+ {
337
+ "loss": 1.157,
338
+ "grad_norm": 3.8281562328338623,
339
+ "learning_rate": 0.00017683015426791566,
340
+ "epoch": 0.2707808564231738,
341
+ "step": 430
342
+ },
343
+ {
344
+ "loss": 1.1627,
345
+ "grad_norm": 5.250668048858643,
346
+ "learning_rate": 0.00017546172308912213,
347
+ "epoch": 0.2770780856423174,
348
+ "step": 440
349
+ },
350
+ {
351
+ "loss": 1.1441,
352
+ "grad_norm": 5.779282569885254,
353
+ "learning_rate": 0.0001740596559006929,
354
+ "epoch": 0.28337531486146095,
355
+ "step": 450
356
+ },
357
+ {
358
+ "loss": 1.1383,
359
+ "grad_norm": 4.290262222290039,
360
+ "learning_rate": 0.00017262457765450457,
361
+ "epoch": 0.28967254408060455,
362
+ "step": 460
363
+ },
364
+ {
365
+ "loss": 1.212,
366
+ "grad_norm": 4.313045024871826,
367
+ "learning_rate": 0.000171157128016652,
368
+ "epoch": 0.2959697732997481,
369
+ "step": 470
370
+ },
371
+ {
372
+ "eval_loss": 1.1923881769180298,
373
+ "eval_runtime": 18.6387,
374
+ "eval_samples_per_second": 27.845,
375
+ "eval_steps_per_second": 6.975,
376
+ "epoch": 0.29848866498740556,
377
+ "step": 474
378
+ },
379
+ {
380
+ "loss": 1.0729,
381
+ "grad_norm": 3.9757778644561768,
382
+ "learning_rate": 0.00016965796108232598,
383
+ "epoch": 0.3022670025188917,
384
+ "step": 480
385
+ },
386
+ {
387
+ "loss": 1.1157,
388
+ "grad_norm": 5.032901287078857,
389
+ "learning_rate": 0.00016812774508425933,
390
+ "epoch": 0.30856423173803527,
391
+ "step": 490
392
+ },
393
+ {
394
+ "loss": 1.1882,
395
+ "grad_norm": 4.437510013580322,
396
+ "learning_rate": 0.00016656716209487174,
397
+ "epoch": 0.3148614609571788,
398
+ "step": 500
399
+ },
400
+ {
401
+ "loss": 1.2293,
402
+ "grad_norm": 4.184134006500244,
403
+ "learning_rate": 0.00016497690772224522,
404
+ "epoch": 0.3211586901763224,
405
+ "step": 510
406
+ },
407
+ {
408
+ "loss": 1.0951,
409
+ "grad_norm": 7.480218410491943,
410
+ "learning_rate": 0.00016335769080006674,
411
+ "epoch": 0.327455919395466,
412
+ "step": 520
413
+ },
414
+ {
415
+ "loss": 1.2974,
416
+ "grad_norm": 3.937443256378174,
417
+ "learning_rate": 0.00016171023307167545,
418
+ "epoch": 0.3337531486146096,
419
+ "step": 530
420
+ },
421
+ {
422
+ "loss": 1.0903,
423
+ "grad_norm": 5.319689750671387,
424
+ "learning_rate": 0.00016003526886835553,
425
+ "epoch": 0.34005037783375314,
426
+ "step": 540
427
+ },
428
+ {
429
+ "loss": 1.1943,
430
+ "grad_norm": 4.3774733543396,
431
+ "learning_rate": 0.00015833354478201844,
432
+ "epoch": 0.34634760705289674,
433
+ "step": 550
434
+ },
435
+ {
436
+ "eval_loss": 1.1556044816970825,
437
+ "eval_runtime": 18.5813,
438
+ "eval_samples_per_second": 27.931,
439
+ "eval_steps_per_second": 6.996,
440
+ "epoch": 0.3482367758186398,
441
+ "step": 553
442
+ },
443
+ {
444
+ "loss": 1.1484,
445
+ "grad_norm": 4.87918758392334,
446
+ "learning_rate": 0.00015660581933241993,
447
+ "epoch": 0.3526448362720403,
448
+ "step": 560
449
+ },
450
+ {
451
+ "loss": 1.1569,
452
+ "grad_norm": 5.953218936920166,
453
+ "learning_rate": 0.00015485286262906044,
454
+ "epoch": 0.3589420654911839,
455
+ "step": 570
456
+ },
457
+ {
458
+ "loss": 1.1309,
459
+ "grad_norm": 3.7059683799743652,
460
+ "learning_rate": 0.00015307545602791955,
461
+ "epoch": 0.36523929471032746,
462
+ "step": 580
463
+ },
464
+ {
465
+ "loss": 1.1818,
466
+ "grad_norm": 3.702540159225464,
467
+ "learning_rate": 0.00015127439178317745,
468
+ "epoch": 0.371536523929471,
469
+ "step": 590
470
+ },
471
+ {
472
+ "loss": 1.0851,
473
+ "grad_norm": 3.8618204593658447,
474
+ "learning_rate": 0.00014945047269407876,
475
+ "epoch": 0.3778337531486146,
476
+ "step": 600
477
+ },
478
+ {
479
+ "loss": 1.2035,
480
+ "grad_norm": 3.884420871734619,
481
+ "learning_rate": 0.0001476045117470958,
482
+ "epoch": 0.38413098236775817,
483
+ "step": 610
484
+ },
485
+ {
486
+ "loss": 1.1274,
487
+ "grad_norm": 4.641972064971924,
488
+ "learning_rate": 0.0001457373317535515,
489
+ "epoch": 0.3904282115869018,
490
+ "step": 620
491
+ },
492
+ {
493
+ "loss": 1.1741,
494
+ "grad_norm": 3.4928109645843506,
495
+ "learning_rate": 0.00014384976498286252,
496
+ "epoch": 0.3967254408060453,
497
+ "step": 630
498
+ },
499
+ {
500
+ "eval_loss": 1.1360982656478882,
501
+ "eval_runtime": 18.5975,
502
+ "eval_samples_per_second": 27.907,
503
+ "eval_steps_per_second": 6.99,
504
+ "epoch": 0.3979848866498741,
505
+ "step": 632
506
+ },
507
+ {
508
+ "loss": 1.0498,
509
+ "grad_norm": 5.098562240600586,
510
+ "learning_rate": 0.00014194265279156675,
511
+ "epoch": 0.40302267002518893,
512
+ "step": 640
513
+ },
514
+ {
515
+ "loss": 1.0625,
516
+ "grad_norm": 4.585139274597168,
517
+ "learning_rate": 0.00014001684524830057,
518
+ "epoch": 0.4093198992443325,
519
+ "step": 650
520
+ },
521
+ {
522
+ "loss": 1.3131,
523
+ "grad_norm": 4.212075233459473,
524
+ "learning_rate": 0.0001380732007548925,
525
+ "epoch": 0.4156171284634761,
526
+ "step": 660
527
+ },
528
+ {
529
+ "loss": 1.0974,
530
+ "grad_norm": 2.9548299312591553,
531
+ "learning_rate": 0.00013611258566374257,
532
+ "epoch": 0.42191435768261965,
533
+ "step": 670
534
+ },
535
+ {
536
+ "loss": 1.1007,
537
+ "grad_norm": 7.119769096374512,
538
+ "learning_rate": 0.00013413587389165784,
539
+ "epoch": 0.4282115869017632,
540
+ "step": 680
541
+ },
542
+ {
543
+ "loss": 1.1405,
544
+ "grad_norm": 3.9440793991088867,
545
+ "learning_rate": 0.00013214394653031616,
546
+ "epoch": 0.4345088161209068,
547
+ "step": 690
548
+ },
549
+ {
550
+ "loss": 1.1412,
551
+ "grad_norm": 3.7775681018829346,
552
+ "learning_rate": 0.000130137691453532,
553
+ "epoch": 0.44080604534005036,
554
+ "step": 700
555
+ },
556
+ {
557
+ "loss": 1.0888,
558
+ "grad_norm": 3.689419984817505,
559
+ "learning_rate": 0.0001281180029214988,
560
+ "epoch": 0.44710327455919396,
561
+ "step": 710
562
+ },
563
+ {
564
+ "eval_loss": 1.1331086158752441,
565
+ "eval_runtime": 18.9998,
566
+ "eval_samples_per_second": 27.316,
567
+ "eval_steps_per_second": 6.842,
568
+ "epoch": 0.4477329974811083,
569
+ "step": 711
570
+ },
571
+ {
572
+ "loss": 1.0959,
573
+ "grad_norm": 4.162503242492676,
574
+ "learning_rate": 0.00012608578118218535,
575
+ "epoch": 0.4534005037783375,
576
+ "step": 720
577
+ },
578
+ {
579
+ "loss": 1.0881,
580
+ "grad_norm": 4.154449939727783,
581
+ "learning_rate": 0.00012404193207006246,
582
+ "epoch": 0.4596977329974811,
583
+ "step": 730
584
+ },
585
+ {
586
+ "loss": 1.1664,
587
+ "grad_norm": 3.9392330646514893,
588
+ "learning_rate": 0.00012198736660234009,
589
+ "epoch": 0.4659949622166247,
590
+ "step": 740
591
+ },
592
+ {
593
+ "loss": 1.2067,
594
+ "grad_norm": 3.568701982498169,
595
+ "learning_rate": 0.0001199230005728942,
596
+ "epoch": 0.4722921914357683,
597
+ "step": 750
598
+ },
599
+ {
600
+ "loss": 1.085,
601
+ "grad_norm": 4.274581432342529,
602
+ "learning_rate": 0.00011784975414406426,
603
+ "epoch": 0.47858942065491183,
604
+ "step": 760
605
+ },
606
+ {
607
+ "loss": 1.1039,
608
+ "grad_norm": 4.135290145874023,
609
+ "learning_rate": 0.00011576855143650371,
610
+ "epoch": 0.48488664987405544,
611
+ "step": 770
612
+ },
613
+ {
614
+ "loss": 1.0403,
615
+ "grad_norm": 3.5335512161254883,
616
+ "learning_rate": 0.00011368032011726635,
617
+ "epoch": 0.491183879093199,
618
+ "step": 780
619
+ },
620
+ {
621
+ "loss": 1.1489,
622
+ "grad_norm": 3.251199722290039,
623
+ "learning_rate": 0.00011158599098631137,
624
+ "epoch": 0.49748110831234255,
625
+ "step": 790
626
+ },
627
+ {
628
+ "eval_loss": 1.1518590450286865,
629
+ "eval_runtime": 18.5131,
630
+ "eval_samples_per_second": 28.034,
631
+ "eval_steps_per_second": 7.022,
632
+ "epoch": 0.49748110831234255,
633
+ "step": 790
634
+ },
635
+ {
636
+ "loss": 1.0229,
637
+ "grad_norm": 3.9981064796447754,
638
+ "learning_rate": 0.00010948649756161246,
639
+ "epoch": 0.5037783375314862,
640
+ "step": 800
641
+ },
642
+ {
643
+ "loss": 1.0989,
644
+ "grad_norm": 4.600470066070557,
645
+ "learning_rate": 0.00010738277566305514,
646
+ "epoch": 0.5100755667506297,
647
+ "step": 810
648
+ },
649
+ {
650
+ "loss": 1.0495,
651
+ "grad_norm": 3.7388057708740234,
652
+ "learning_rate": 0.0001052757629953082,
653
+ "epoch": 0.5163727959697733,
654
+ "step": 820
655
+ },
656
+ {
657
+ "loss": 1.0471,
658
+ "grad_norm": 4.045319557189941,
659
+ "learning_rate": 0.00010316639872985472,
660
+ "epoch": 0.5226700251889169,
661
+ "step": 830
662
+ },
663
+ {
664
+ "loss": 1.1542,
665
+ "grad_norm": 3.8340413570404053,
666
+ "learning_rate": 0.00010105562308636973,
667
+ "epoch": 0.5289672544080605,
668
+ "step": 840
669
+ },
670
+ {
671
+ "loss": 1.0944,
672
+ "grad_norm": 3.6133155822753906,
673
+ "learning_rate": 9.894437691363028e-05,
674
+ "epoch": 0.535264483627204,
675
+ "step": 850
676
+ },
677
+ {
678
+ "loss": 1.0703,
679
+ "grad_norm": 3.6610817909240723,
680
+ "learning_rate": 9.683360127014529e-05,
681
+ "epoch": 0.5415617128463476,
682
+ "step": 860
683
+ },
684
+ {
685
+ "eval_loss": 1.1126927137374878,
686
+ "eval_runtime": 18.6703,
687
+ "eval_samples_per_second": 27.798,
688
+ "eval_steps_per_second": 6.963,
689
+ "epoch": 0.5472292191435768,
690
+ "step": 869
691
+ },
692
+ {
693
+ "loss": 1.0068,
694
+ "grad_norm": 3.7706260681152344,
695
+ "learning_rate": 9.47242370046918e-05,
696
+ "epoch": 0.5478589420654912,
697
+ "step": 870
698
+ },
699
+ {
700
+ "loss": 1.0582,
701
+ "grad_norm": 4.216574192047119,
702
+ "learning_rate": 9.261722433694485e-05,
703
+ "epoch": 0.5541561712846348,
704
+ "step": 880
705
+ },
706
+ {
707
+ "loss": 1.091,
708
+ "grad_norm": 4.351677894592285,
709
+ "learning_rate": 9.051350243838756e-05,
710
+ "epoch": 0.5604534005037783,
711
+ "step": 890
712
+ },
713
+ {
714
+ "loss": 1.0095,
715
+ "grad_norm": 3.1938083171844482,
716
+ "learning_rate": 8.841400901368864e-05,
717
+ "epoch": 0.5667506297229219,
718
+ "step": 900
719
+ },
720
+ {
721
+ "loss": 0.9896,
722
+ "grad_norm": 3.466660976409912,
723
+ "learning_rate": 8.631967988273366e-05,
724
+ "epoch": 0.5730478589420654,
725
+ "step": 910
726
+ },
727
+ {
728
+ "loss": 0.9416,
729
+ "grad_norm": 4.237432479858398,
730
+ "learning_rate": 8.423144856349631e-05,
731
+ "epoch": 0.5793450881612091,
732
+ "step": 920
733
+ },
734
+ {
735
+ "loss": 1.1146,
736
+ "grad_norm": 2.8684630393981934,
737
+ "learning_rate": 8.215024585593577e-05,
738
+ "epoch": 0.5856423173803527,
739
+ "step": 930
740
+ },
741
+ {
742
+ "loss": 1.0055,
743
+ "grad_norm": 3.612492322921753,
744
+ "learning_rate": 8.007699942710581e-05,
745
+ "epoch": 0.5919395465994962,
746
+ "step": 940
747
+ },
748
+ {
749
+ "eval_loss": 1.072987675666809,
750
+ "eval_runtime": 18.7293,
751
+ "eval_samples_per_second": 27.711,
752
+ "eval_steps_per_second": 6.941,
753
+ "epoch": 0.5969773299748111,
754
+ "step": 948
755
+ },
756
+ {
757
+ "loss": 1.0303,
758
+ "grad_norm": 4.036654472351074,
759
+ "learning_rate": 7.801263339765994e-05,
760
+ "epoch": 0.5982367758186398,
761
+ "step": 950
762
+ },
763
+ {
764
+ "loss": 1.0264,
765
+ "grad_norm": 3.446193218231201,
766
+ "learning_rate": 7.595806792993759e-05,
767
+ "epoch": 0.6045340050377834,
768
+ "step": 960
769
+ },
770
+ {
771
+ "loss": 1.1178,
772
+ "grad_norm": 4.124652862548828,
773
+ "learning_rate": 7.391421881781468e-05,
774
+ "epoch": 0.610831234256927,
775
+ "step": 970
776
+ },
777
+ {
778
+ "loss": 1.0875,
779
+ "grad_norm": 3.254272699356079,
780
+ "learning_rate": 7.188199707850122e-05,
781
+ "epoch": 0.6171284634760705,
782
+ "step": 980
783
+ },
784
+ {
785
+ "loss": 1.0247,
786
+ "grad_norm": 4.263841152191162,
787
+ "learning_rate": 6.986230854646804e-05,
788
+ "epoch": 0.6234256926952141,
789
+ "step": 990
790
+ },
791
+ {
792
+ "loss": 1.0394,
793
+ "grad_norm": 4.2250447273254395,
794
+ "learning_rate": 6.785605346968386e-05,
795
+ "epoch": 0.6297229219143576,
796
+ "step": 1000
797
+ },
798
+ {
799
+ "loss": 1.058,
800
+ "grad_norm": 4.4213128089904785,
801
+ "learning_rate": 6.586412610834221e-05,
802
+ "epoch": 0.6360201511335013,
803
+ "step": 1010
804
+ },
805
+ {
806
+ "loss": 1.0354,
807
+ "grad_norm": 3.6707351207733154,
808
+ "learning_rate": 6.388741433625746e-05,
809
+ "epoch": 0.6423173803526449,
810
+ "step": 1020
811
+ },
812
+ {
813
+ "eval_loss": 1.0599287748336792,
814
+ "eval_runtime": 18.7825,
815
+ "eval_samples_per_second": 27.632,
816
+ "eval_steps_per_second": 6.921,
817
+ "epoch": 0.6467254408060453,
818
+ "step": 1027
819
+ },
820
+ {
821
+ "loss": 1.0219,
822
+ "grad_norm": 3.590726613998413,
823
+ "learning_rate": 6.192679924510752e-05,
824
+ "epoch": 0.6486146095717884,
825
+ "step": 1030
826
+ },
827
+ {
828
+ "loss": 0.9563,
829
+ "grad_norm": 2.9363389015197754,
830
+ "learning_rate": 5.998315475169942e-05,
831
+ "epoch": 0.654911838790932,
832
+ "step": 1040
833
+ },
834
+ {
835
+ "loss": 1.0477,
836
+ "grad_norm": 3.508352279663086,
837
+ "learning_rate": 5.8057347208433266e-05,
838
+ "epoch": 0.6612090680100756,
839
+ "step": 1050
840
+ },
841
+ {
842
+ "loss": 0.9836,
843
+ "grad_norm": 3.6980814933776855,
844
+ "learning_rate": 5.615023501713752e-05,
845
+ "epoch": 0.6675062972292192,
846
+ "step": 1060
847
+ },
848
+ {
849
+ "loss": 1.1597,
850
+ "grad_norm": 4.591259002685547,
851
+ "learning_rate": 5.4262668246448475e-05,
852
+ "epoch": 0.6738035264483627,
853
+ "step": 1070
854
+ },
855
+ {
856
+ "loss": 1.0229,
857
+ "grad_norm": 4.845359802246094,
858
+ "learning_rate": 5.239548825290419e-05,
859
+ "epoch": 0.6801007556675063,
860
+ "step": 1080
861
+ },
862
+ {
863
+ "loss": 0.9992,
864
+ "grad_norm": 3.658125400543213,
865
+ "learning_rate": 5.0549527305921265e-05,
866
+ "epoch": 0.6863979848866498,
867
+ "step": 1090
868
+ },
869
+ {
870
+ "loss": 1.0752,
871
+ "grad_norm": 4.482686996459961,
872
+ "learning_rate": 4.872560821682256e-05,
873
+ "epoch": 0.6926952141057935,
874
+ "step": 1100
875
+ },
876
+ {
877
+ "eval_loss": 1.054300308227539,
878
+ "eval_runtime": 18.6514,
879
+ "eval_samples_per_second": 27.826,
880
+ "eval_steps_per_second": 6.97,
881
+ "epoch": 0.6964735516372796,
882
+ "step": 1106
883
+ },
884
+ {
885
+ "loss": 1.1193,
886
+ "grad_norm": 4.97657585144043,
887
+ "learning_rate": 4.692454397208048e-05,
888
+ "epoch": 0.698992443324937,
889
+ "step": 1110
890
+ },
891
+ {
892
+ "loss": 1.0533,
893
+ "grad_norm": 3.9969918727874756,
894
+ "learning_rate": 4.5147137370939595e-05,
895
+ "epoch": 0.7052896725440806,
896
+ "step": 1120
897
+ },
898
+ {
899
+ "loss": 0.9617,
900
+ "grad_norm": 4.142766952514648,
901
+ "learning_rate": 4.339418066758008e-05,
902
+ "epoch": 0.7115869017632241,
903
+ "step": 1130
904
+ },
905
+ {
906
+ "loss": 1.094,
907
+ "grad_norm": 3.6601150035858154,
908
+ "learning_rate": 4.166645521798157e-05,
909
+ "epoch": 0.7178841309823678,
910
+ "step": 1140
911
+ },
912
+ {
913
+ "loss": 0.9629,
914
+ "grad_norm": 3.6469886302948,
915
+ "learning_rate": 3.99647311316445e-05,
916
+ "epoch": 0.7241813602015114,
917
+ "step": 1150
918
+ },
919
+ {
920
+ "loss": 1.0169,
921
+ "grad_norm": 3.6334595680236816,
922
+ "learning_rate": 3.828976692832458e-05,
923
+ "epoch": 0.7304785894206549,
924
+ "step": 1160
925
+ },
926
+ {
927
+ "loss": 0.9622,
928
+ "grad_norm": 3.6162374019622803,
929
+ "learning_rate": 3.6642309199933275e-05,
930
+ "epoch": 0.7367758186397985,
931
+ "step": 1170
932
+ },
933
+ {
934
+ "loss": 1.0783,
935
+ "grad_norm": 3.4192535877227783,
936
+ "learning_rate": 3.502309227775482e-05,
937
+ "epoch": 0.743073047858942,
938
+ "step": 1180
939
+ },
940
+ {
941
+ "eval_loss": 1.046454906463623,
942
+ "eval_runtime": 18.8296,
943
+ "eval_samples_per_second": 27.563,
944
+ "eval_steps_per_second": 6.904,
945
+ "epoch": 0.7462216624685138,
946
+ "step": 1185
947
+ },
948
+ {
949
+ "loss": 0.9546,
950
+ "grad_norm": 3.3051228523254395,
951
+ "learning_rate": 3.343283790512829e-05,
952
+ "epoch": 0.7493702770780857,
953
+ "step": 1190
954
+ },
955
+ {
956
+ "loss": 0.9628,
957
+ "grad_norm": 3.579636812210083,
958
+ "learning_rate": 3.1872254915740686e-05,
959
+ "epoch": 0.7556675062972292,
960
+ "step": 1200
961
+ },
962
+ {
963
+ "loss": 1.1041,
964
+ "grad_norm": 4.326976776123047,
965
+ "learning_rate": 3.0342038917674064e-05,
966
+ "epoch": 0.7619647355163728,
967
+ "step": 1210
968
+ },
969
+ {
970
+ "loss": 1.1097,
971
+ "grad_norm": 4.363333225250244,
972
+ "learning_rate": 2.8842871983347998e-05,
973
+ "epoch": 0.7682619647355163,
974
+ "step": 1220
975
+ },
976
+ {
977
+ "loss": 0.9976,
978
+ "grad_norm": 3.557300567626953,
979
+ "learning_rate": 2.7375422345495448e-05,
980
+ "epoch": 0.77455919395466,
981
+ "step": 1230
982
+ },
983
+ {
984
+ "loss": 1.0998,
985
+ "grad_norm": 4.066524982452393,
986
+ "learning_rate": 2.5940344099307146e-05,
987
+ "epoch": 0.7808564231738035,
988
+ "step": 1240
989
+ },
990
+ {
991
+ "loss": 1.0098,
992
+ "grad_norm": 4.808539390563965,
993
+ "learning_rate": 2.45382769108779e-05,
994
+ "epoch": 0.7871536523929471,
995
+ "step": 1250
996
+ },
997
+ {
998
+ "loss": 1.0634,
999
+ "grad_norm": 3.593735456466675,
1000
+ "learning_rate": 2.3169845732084338e-05,
1001
+ "epoch": 0.7934508816120907,
1002
+ "step": 1260
1003
+ },
1004
+ {
1005
+ "eval_loss": 1.032251238822937,
1006
+ "eval_runtime": 18.7335,
1007
+ "eval_samples_per_second": 27.704,
1008
+ "eval_steps_per_second": 6.939,
1009
+ "epoch": 0.7959697732997482,
1010
+ "step": 1264
1011
+ },
1012
+ {
1013
+ "loss": 1.0001,
1014
+ "grad_norm": 3.325232744216919,
1015
+ "learning_rate": 2.1835660522021417e-05,
1016
+ "epoch": 0.7997481108312342,
1017
+ "step": 1270
1018
+ },
1019
+ {
1020
+ "loss": 0.9349,
1021
+ "grad_norm": 4.100666046142578,
1022
+ "learning_rate": 2.0536315975121544e-05,
1023
+ "epoch": 0.8060453400503779,
1024
+ "step": 1280
1025
+ },
1026
+ {
1027
+ "loss": 0.989,
1028
+ "grad_norm": 3.3437387943267822,
1029
+ "learning_rate": 1.927239125607788e-05,
1030
+ "epoch": 0.8123425692695214,
1031
+ "step": 1290
1032
+ },
1033
+ {
1034
+ "loss": 1.0185,
1035
+ "grad_norm": 3.979337453842163,
1036
+ "learning_rate": 1.8044449741689794e-05,
1037
+ "epoch": 0.818639798488665,
1038
+ "step": 1300
1039
+ },
1040
+ {
1041
+ "loss": 1.0439,
1042
+ "grad_norm": 6.503814697265625,
1043
+ "learning_rate": 1.6853038769745467e-05,
1044
+ "epoch": 0.8249370277078085,
1045
+ "step": 1310
1046
+ },
1047
+ {
1048
+ "loss": 0.9196,
1049
+ "grad_norm": 3.8204808235168457,
1050
+ "learning_rate": 1.5698689395053835e-05,
1051
+ "epoch": 0.8312342569269522,
1052
+ "step": 1320
1053
+ },
1054
+ {
1055
+ "loss": 1.1313,
1056
+ "grad_norm": 4.812267303466797,
1057
+ "learning_rate": 1.4581916152734132e-05,
1058
+ "epoch": 0.8375314861460957,
1059
+ "step": 1330
1060
+ },
1061
+ {
1062
+ "loss": 0.98,
1063
+ "grad_norm": 3.7348341941833496,
1064
+ "learning_rate": 1.3503216828869192e-05,
1065
+ "epoch": 0.8438287153652393,
1066
+ "step": 1340
1067
+ },
1068
+ {
1069
+ "eval_loss": 1.0260143280029297,
1070
+ "eval_runtime": 18.6543,
1071
+ "eval_samples_per_second": 27.822,
1072
+ "eval_steps_per_second": 6.969,
1073
+ "epoch": 0.8457178841309824,
1074
+ "step": 1343
1075
+ },
1076
+ {
1077
+ "loss": 1.1145,
1078
+ "grad_norm": 5.438048362731934,
1079
+ "learning_rate": 1.2463072238624185e-05,
1080
+ "epoch": 0.8501259445843828,
1081
+ "step": 1350
1082
+ },
1083
+ {
1084
+ "loss": 0.9843,
1085
+ "grad_norm": 3.407062292098999,
1086
+ "learning_rate": 1.146194601192998e-05,
1087
+ "epoch": 0.8564231738035264,
1088
+ "step": 1360
1089
+ },
1090
+ {
1091
+ "loss": 1.0344,
1092
+ "grad_norm": 4.0413713455200195,
1093
+ "learning_rate": 1.0500284386826597e-05,
1094
+ "epoch": 0.8627204030226701,
1095
+ "step": 1370
1096
+ },
1097
+ {
1098
+ "loss": 1.0159,
1099
+ "grad_norm": 4.10891580581665,
1100
+ "learning_rate": 9.578516010558847e-06,
1101
+ "epoch": 0.8690176322418136,
1102
+ "step": 1380
1103
+ },
1104
+ {
1105
+ "loss": 1.0693,
1106
+ "grad_norm": 3.644531011581421,
1107
+ "learning_rate": 8.697051748512708e-06,
1108
+ "epoch": 0.8753148614609572,
1109
+ "step": 1390
1110
+ },
1111
+ {
1112
+ "loss": 1.0402,
1113
+ "grad_norm": 3.6608030796051025,
1114
+ "learning_rate": 7.856284501077926e-06,
1115
+ "epoch": 0.8816120906801007,
1116
+ "step": 1400
1117
+ },
1118
+ {
1119
+ "loss": 0.9965,
1120
+ "grad_norm": 3.7804226875305176,
1121
+ "learning_rate": 7.056589028518068e-06,
1122
+ "epoch": 0.8879093198992444,
1123
+ "step": 1410
1124
+ },
1125
+ {
1126
+ "loss": 1.0403,
1127
+ "grad_norm": 4.699734687805176,
1128
+ "learning_rate": 6.2983217839264756e-06,
1129
+ "epoch": 0.8942065491183879,
1130
+ "step": 1420
1131
+ },
1132
+ {
1133
+ "eval_loss": 1.0198695659637451,
1134
+ "eval_runtime": 18.7497,
1135
+ "eval_samples_per_second": 27.68,
1136
+ "eval_steps_per_second": 6.933,
1137
+ "epoch": 0.8954659949622166,
1138
+ "step": 1422
1139
+ },
1140
+ {
1141
+ "loss": 0.997,
1142
+ "grad_norm": 4.407861232757568,
1143
+ "learning_rate": 5.581820754342137e-06,
1144
+ "epoch": 0.9005037783375315,
1145
+ "step": 1430
1146
+ },
1147
+ {
1148
+ "loss": 0.9397,
1149
+ "grad_norm": 3.5445852279663086,
1150
+ "learning_rate": 4.907405310096769e-06,
1151
+ "epoch": 0.906801007556675,
1152
+ "step": 1440
1153
+ },
1154
+ {
1155
+ "loss": 0.9893,
1156
+ "grad_norm": 4.443542003631592,
1157
+ "learning_rate": 4.275376062459946e-06,
1158
+ "epoch": 0.9130982367758187,
1159
+ "step": 1450
1160
+ },
1161
+ {
1162
+ "loss": 0.9178,
1163
+ "grad_norm": 3.4546210765838623,
1164
+ "learning_rate": 3.6860147296457816e-06,
1165
+ "epoch": 0.9193954659949622,
1166
+ "step": 1460
1167
+ },
1168
+ {
1169
+ "loss": 0.9573,
1170
+ "grad_norm": 4.19729471206665,
1171
+ "learning_rate": 3.1395840112410034e-06,
1172
+ "epoch": 0.9256926952141058,
1173
+ "step": 1470
1174
+ },
1175
+ {
1176
+ "loss": 1.0305,
1177
+ "grad_norm": 3.5583102703094482,
1178
+ "learning_rate": 2.636327471110289e-06,
1179
+ "epoch": 0.9319899244332494,
1180
+ "step": 1480
1181
+ },
1182
+ {
1183
+ "loss": 0.9903,
1184
+ "grad_norm": 4.989317417144775,
1185
+ "learning_rate": 2.1764694288310184e-06,
1186
+ "epoch": 0.9382871536523929,
1187
+ "step": 1490
1188
+ },
1189
+ {
1190
+ "loss": 0.9876,
1191
+ "grad_norm": 2.8516058921813965,
1192
+ "learning_rate": 1.7602148597060108e-06,
1193
+ "epoch": 0.9445843828715366,
1194
+ "step": 1500
1195
+ },
1196
+ {
1197
+ "eval_loss": 1.019551396369934,
1198
+ "eval_runtime": 18.5676,
1199
+ "eval_samples_per_second": 27.952,
1200
+ "eval_steps_per_second": 7.001,
1201
+ "epoch": 0.9452141057934509,
1202
+ "step": 1501
1203
+ },
1204
+ {
1205
+ "loss": 0.9417,
1206
+ "grad_norm": 3.9421305656433105,
1207
+ "learning_rate": 1.3877493033986223e-06,
1208
+ "epoch": 0.9508816120906801,
1209
+ "step": 1510
1210
+ },
1211
+ {
1212
+ "loss": 0.9706,
1213
+ "grad_norm": 4.05004358291626,
1214
+ "learning_rate": 1.0592387812310311e-06,
1215
+ "epoch": 0.9571788413098237,
1216
+ "step": 1520
1217
+ },
1218
+ {
1219
+ "loss": 0.9875,
1220
+ "grad_norm": 3.352207660675049,
1221
+ "learning_rate": 7.748297221825573e-07,
1222
+ "epoch": 0.9634760705289672,
1223
+ "step": 1530
1224
+ },
1225
+ {
1226
+ "loss": 0.9976,
1227
+ "grad_norm": 3.546070098876953,
1228
+ "learning_rate": 5.346488976209596e-07,
1229
+ "epoch": 0.9697732997481109,
1230
+ "step": 1540
1231
+ },
1232
+ {
1233
+ "loss": 0.9848,
1234
+ "grad_norm": 4.3622822761535645,
1235
+ "learning_rate": 3.3880336479590325e-07,
1236
+ "epoch": 0.9760705289672544,
1237
+ "step": 1550
1238
+ },
1239
+ {
1240
+ "loss": 1.0212,
1241
+ "grad_norm": 3.776240110397339,
1242
+ "learning_rate": 1.8738041911969818e-07,
1243
+ "epoch": 0.982367758186398,
1244
+ "step": 1560
1245
+ },
1246
+ {
1247
+ "loss": 1.0542,
1248
+ "grad_norm": 4.287234306335449,
1249
+ "learning_rate": 8.044755525652425e-08,
1250
+ "epoch": 0.9886649874055415,
1251
+ "step": 1570
1252
+ },
1253
+ {
1254
+ "loss": 0.988,
1255
+ "grad_norm": 3.729553461074829,
1256
+ "learning_rate": 1.8052437037707758e-08,
1257
+ "epoch": 0.9949622166246851,
1258
+ "step": 1580
1259
+ },
1260
+ {
1261
+ "eval_loss": 1.0195945501327515,
1262
+ "eval_runtime": 18.5885,
1263
+ "eval_samples_per_second": 27.921,
1264
+ "eval_steps_per_second": 6.994,
1265
+ "epoch": 0.9949622166246851,
1266
+ "step": 1580
1267
+ },
1268
+ {
1269
+ "train_runtime": 2468.3919,
1270
+ "train_samples_per_second": 10.293,
1271
+ "train_steps_per_second": 0.643,
1272
+ "total_flos": 7.792418530588262e+16,
1273
+ "train_loss": 1.1395688813639528,
1274
+ "epoch": 1.0,
1275
+ "step": 1588
1276
+ }
1277
+ ]
train/training_loss.png ADDED
train/validation_loss.png ADDED