kiolPL commited on
Commit
28fac09
·
1 Parent(s): 9c5fba6

Upload NaturaVision QLoRA adapter

Browse files
README.md CHANGED
@@ -1,3 +1,82 @@
1
- ---
2
- license: cc-by-nc-4.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen3.5-4B
3
+ library_name: peft
4
+ pipeline_tag: image-text-to-text
5
+ license: cc-by-nc-4.0
6
+ tags:
7
+ - qwen3_5
8
+ - peft
9
+ - lora
10
+ - qlora
11
+ - image-text-to-text
12
+ - species-classification
13
+ - naturavision
14
+ ---
15
+
16
+ # NaturaVision Qwen3.5 4B QLoRA Adapter
17
+
18
+ This repository contains the PEFT LoRA adapter trained for NaturaVision, a forest plant and fungi recognition prototype. The adapter is intended to be loaded on top of `Qwen/Qwen3.5-4B`.
19
+
20
+ The model was trained to return a minimal JSON response:
21
+
22
+ ```json
23
+ {"label_id":"PLANT_01"}
24
+ ```
25
+
26
+ The full species metadata is provided in `labels.json` and `species_manifest.csv`.
27
+
28
+ ## Task
29
+
30
+ The adapter classifies an input image into a fixed taxonomy:
31
+
32
+ - 20 forest plant classes,
33
+ - 20 forest fungi classes,
34
+ - `unknown` for organisms outside the supported taxonomy or ambiguous images.
35
+
36
+ ## Files
37
+
38
+ - `adapter_model.safetensors` - PEFT LoRA adapter weights.
39
+ - `adapter_config.json` - PEFT adapter configuration.
40
+ - `additional_config.json` - extra LoRA+/training configuration saved by the training framework.
41
+ - `args.json` - training run arguments.
42
+ - `trainer_state.json` - trainer state for the selected checkpoint.
43
+ - `labels.json` - label metadata used by the application and evaluator.
44
+ - `species_manifest.csv` - compact public taxonomy manifest.
45
+
46
+ ## Evaluation Summary
47
+
48
+ The selected checkpoint was evaluated with generative inference on a held-out test split.
49
+
50
+ | Metric | Base Qwen3.5-4B | NaturaVision adapter |
51
+ |---|---:|---:|
52
+ | Accuracy | 0.4126 | 0.7690 |
53
+ | Macro-F1 known | 0.3290 | 0.7572 |
54
+ | Macro-F1 all | 0.3216 | 0.7594 |
55
+ | Unknown precision | 0.6281 | 0.9562 |
56
+ | Unknown recall | 0.7225 | 0.7572 |
57
+ | Unknown F1 | 0.6720 | 0.8452 |
58
+ | Valid JSON rate | 0.9991 | 1.0000 |
59
+
60
+ Validation and test results were close: validation accuracy was `0.7708`, while test accuracy was `0.7690`.
61
+
62
+ ## Training Setup
63
+
64
+ - Base model: `Qwen/Qwen3.5-4B`
65
+ - Method: 4-bit QLoRA with PEFT LoRA adapters
66
+ - Quantization: bitsandbytes NF4 during training
67
+ - Adapter: rsLoRA, LoRA rank `8`, alpha `32`, dropout `0.05`
68
+ - Framework: ms-swift / PEFT
69
+ - Hardware: NVIDIA RTX 4070 12 GB
70
+ - Selected checkpoint: `checkpoint-1359`
71
+
72
+ ## Limitations
73
+
74
+ This is a project checkpoint, not a production species identification system. The model is trained on a closed taxonomy and should return `unknown` for images outside that scope. It should not be used as the only source for safety-critical mushroom or plant identification decisions.
75
+
76
+ ## Intended Use
77
+
78
+ The adapter is intended for the NaturaVision Android prototype and for reproducible evaluation of the project checkpoint. Downstream applications should map the generated `label_id` through `labels.json` instead of asking the model to generate long species descriptions.
79
+
80
+ ## License
81
+
82
+ The base model `Qwen/Qwen3.5-4B` is governed by its Apache-2.0 license. This adapter is released under CC BY-NC 4.0 because it was trained on an iNaturalist-derived dataset containing CC-BY-NC licensed media.
adapter_config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "Qwen/Qwen3.5-4B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": [],
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 8,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": "^(model\\.visual\\.merger(?=\\.).*\\.(linear_fc1|linear_fc2))$",
32
+ "target_parameters": null,
33
+ "task_type": "CAUSAL_LM",
34
+ "trainable_token_indices": null,
35
+ "use_dora": false,
36
+ "use_qalora": false,
37
+ "use_rslora": true
38
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a661357c29ef4ddebe46c118dcee898cdf76715d2210f57db4a497f8ae1981a
3
+ size 475688
additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": 16.0, "lorap_emb_lr": 1e-06}
args.json ADDED
@@ -0,0 +1,345 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332",
3
+ "per_device_train_batch_size": 1,
4
+ "num_train_epochs": 3.0,
5
+ "max_steps": 1359,
6
+ "learning_rate": 0.0001,
7
+ "lr_scheduler_type": "cosine",
8
+ "lr_scheduler_kwargs": null,
9
+ "warmup_steps": 0,
10
+ "optim": "adamw_torch_fused",
11
+ "optim_args": null,
12
+ "weight_decay": 0.1,
13
+ "adam_beta1": 0.9,
14
+ "adam_beta2": 0.95,
15
+ "adam_epsilon": 1e-08,
16
+ "optim_target_modules": null,
17
+ "gradient_accumulation_steps": 32,
18
+ "average_tokens_across_devices": true,
19
+ "max_grad_norm": 1.0,
20
+ "label_smoothing_factor": 0.0,
21
+ "bf16": true,
22
+ "fp16": false,
23
+ "bf16_full_eval": false,
24
+ "fp16_full_eval": false,
25
+ "tf32": null,
26
+ "gradient_checkpointing": true,
27
+ "gradient_checkpointing_kwargs": null,
28
+ "torch_compile": false,
29
+ "torch_compile_backend": null,
30
+ "torch_compile_mode": null,
31
+ "use_liger_kernel": false,
32
+ "liger_kernel_config": null,
33
+ "use_cache": false,
34
+ "neftune_noise_alpha": null,
35
+ "torch_empty_cache_steps": null,
36
+ "auto_find_batch_size": false,
37
+ "logging_strategy": "steps",
38
+ "logging_steps": 5,
39
+ "logging_first_step": true,
40
+ "log_on_each_node": true,
41
+ "logging_nan_inf_filter": true,
42
+ "include_num_input_tokens_seen": false,
43
+ "log_level": "passive",
44
+ "log_level_replica": "warning",
45
+ "disable_tqdm": null,
46
+ "report_to": [
47
+ "tensorboard"
48
+ ],
49
+ "run_name": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332",
50
+ "project": "huggingface",
51
+ "trackio_space_id": "trackio",
52
+ "eval_strategy": "steps",
53
+ "eval_steps": 200.0,
54
+ "eval_delay": 0,
55
+ "per_device_eval_batch_size": 1,
56
+ "prediction_loss_only": false,
57
+ "eval_on_start": false,
58
+ "eval_do_concat_batches": true,
59
+ "eval_use_gather_object": false,
60
+ "eval_accumulation_steps": null,
61
+ "include_for_metrics": [],
62
+ "batch_eval_metrics": false,
63
+ "save_only_model": false,
64
+ "save_strategy": "steps",
65
+ "save_steps": 200.0,
66
+ "save_on_each_node": false,
67
+ "save_total_limit": 2,
68
+ "enable_jit_checkpoint": false,
69
+ "push_to_hub": false,
70
+ "hub_token": null,
71
+ "hub_private_repo": null,
72
+ "hub_model_id": null,
73
+ "hub_strategy": "every_save",
74
+ "hub_always_push": false,
75
+ "hub_revision": null,
76
+ "load_best_model_at_end": false,
77
+ "metric_for_best_model": "loss",
78
+ "greater_is_better": false,
79
+ "ignore_data_skip": true,
80
+ "restore_callback_states_from_checkpoint": false,
81
+ "full_determinism": false,
82
+ "seed": 23,
83
+ "data_seed": 42,
84
+ "use_cpu": false,
85
+ "accelerator_config": {
86
+ "dispatch_batches": false
87
+ },
88
+ "parallelism_config": null,
89
+ "dataloader_drop_last": false,
90
+ "dataloader_num_workers": 4,
91
+ "dataloader_pin_memory": true,
92
+ "dataloader_persistent_workers": false,
93
+ "dataloader_prefetch_factor": null,
94
+ "remove_unused_columns": true,
95
+ "label_names": null,
96
+ "train_sampling_strategy": "random",
97
+ "length_column_name": "length",
98
+ "ddp_find_unused_parameters": null,
99
+ "ddp_bucket_cap_mb": null,
100
+ "ddp_broadcast_buffers": null,
101
+ "ddp_backend": null,
102
+ "ddp_timeout": 18000000,
103
+ "fsdp": [],
104
+ "fsdp_config": null,
105
+ "deepspeed": null,
106
+ "debug": null,
107
+ "skip_memory_metrics": true,
108
+ "do_train": false,
109
+ "do_eval": false,
110
+ "do_predict": false,
111
+ "resume_from_checkpoint": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-aligner/v0-20260424-001639/checkpoint-453",
112
+ "warmup_ratio": 0.05,
113
+ "logging_dir": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332/runs",
114
+ "local_rank": -1,
115
+ "sortish_sampler": false,
116
+ "predict_with_generate": false,
117
+ "generation_max_length": null,
118
+ "generation_num_beams": null,
119
+ "generation_config": null,
120
+ "tuner_backend": "peft",
121
+ "vit_gradient_checkpointing": false,
122
+ "router_aux_loss_coef": 0.0,
123
+ "enable_dft_loss": false,
124
+ "enable_channel_loss": false,
125
+ "safe_serialization": true,
126
+ "max_shard_size": "5GB",
127
+ "check_model": true,
128
+ "acc_strategy": "token",
129
+ "train_dataloader_shuffle": true,
130
+ "group_by_length": true,
131
+ "max_epochs": null,
132
+ "aligner_lr": null,
133
+ "vit_lr": null,
134
+ "use_logits_to_keep": null,
135
+ "ds3_gather_for_generation": true,
136
+ "resume_only_model": true,
137
+ "optimizer": "lorap",
138
+ "loss_type": null,
139
+ "eval_metric": null,
140
+ "callbacks": [],
141
+ "early_stop_interval": null,
142
+ "eval_use_evalscope": false,
143
+ "eval_dataset": [],
144
+ "eval_dataset_args": null,
145
+ "eval_limit": null,
146
+ "eval_generation_config": null,
147
+ "extra_eval_args": null,
148
+ "tuner_type": "lora",
149
+ "use_galore": false,
150
+ "galore_target_modules": null,
151
+ "galore_rank": 128,
152
+ "galore_update_proj_gap": 50,
153
+ "galore_scale": 1.0,
154
+ "galore_proj_type": "std",
155
+ "galore_optim_per_parameter": false,
156
+ "galore_with_embedding": false,
157
+ "galore_quantization": false,
158
+ "galore_proj_quant": false,
159
+ "galore_proj_bits": 4,
160
+ "galore_proj_group_size": 256,
161
+ "galore_cos_threshold": 0.4,
162
+ "galore_gamma_proj": 2,
163
+ "galore_queue_size": 5,
164
+ "lisa_activated_layers": 0,
165
+ "lisa_step_interval": 20,
166
+ "use_flash_ckpt": false,
167
+ "use_ray": false,
168
+ "ray_exp_name": null,
169
+ "device_groups": null,
170
+ "model": "Qwen/Qwen3.5-4B",
171
+ "model_type": "qwen3_5",
172
+ "model_revision": null,
173
+ "task_type": "causal_lm",
174
+ "torch_dtype": "bfloat16",
175
+ "attn_impl": "sdpa",
176
+ "experts_impl": null,
177
+ "new_special_tokens": [],
178
+ "num_labels": null,
179
+ "problem_type": null,
180
+ "rope_scaling": null,
181
+ "device_map": null,
182
+ "max_memory": {},
183
+ "max_model_len": null,
184
+ "local_repo_path": null,
185
+ "init_strategy": null,
186
+ "template": "qwen3_5",
187
+ "system": null,
188
+ "max_length": 1280,
189
+ "truncation_strategy": "delete",
190
+ "max_pixels": 150528,
191
+ "agent_template": null,
192
+ "norm_bbox": null,
193
+ "use_chat_template": true,
194
+ "padding_side": "right",
195
+ "padding_free": false,
196
+ "loss_scale": "ignore_empty_think",
197
+ "sequence_parallel_size": 1,
198
+ "template_backend": "swift",
199
+ "response_prefix": null,
200
+ "enable_thinking": null,
201
+ "add_non_thinking_prefix": true,
202
+ "dataset": [
203
+ "/home/kiol/naturavision-data-v2-clean-r1/train.jsonl"
204
+ ],
205
+ "val_dataset": [
206
+ "/home/kiol/naturavision-data-v2-clean-r1/val.jsonl"
207
+ ],
208
+ "cached_dataset": [],
209
+ "cached_val_dataset": [],
210
+ "split_dataset_ratio": 0.0,
211
+ "dataset_num_proc": 4,
212
+ "load_from_cache_file": true,
213
+ "dataset_shuffle": true,
214
+ "val_dataset_shuffle": false,
215
+ "streaming": false,
216
+ "interleave_prob": null,
217
+ "stopping_strategy": "first_exhausted",
218
+ "shuffle_buffer_size": 1000,
219
+ "download_mode": "reuse_dataset_if_exists",
220
+ "columns": {},
221
+ "strict": false,
222
+ "model_name": null,
223
+ "model_author": null,
224
+ "custom_dataset_info": [],
225
+ "quant_method": "bnb",
226
+ "quant_bits": 4,
227
+ "hqq_axis": null,
228
+ "bnb_4bit_compute_dtype": "bfloat16",
229
+ "bnb_4bit_quant_type": "nf4",
230
+ "bnb_4bit_use_double_quant": true,
231
+ "bnb_4bit_quant_storage": null,
232
+ "max_new_tokens": 64,
233
+ "temperature": 0.0,
234
+ "top_k": null,
235
+ "top_p": null,
236
+ "repetition_penalty": null,
237
+ "num_beams": 1,
238
+ "stream": false,
239
+ "stop_words": [],
240
+ "logprobs": false,
241
+ "top_logprobs": null,
242
+ "structured_outputs_regex": null,
243
+ "adapters": [
244
+ "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-aligner/v0-20260424-001639/checkpoint-453"
245
+ ],
246
+ "external_plugins": [],
247
+ "custom_register_path": [],
248
+ "model_kwargs": {},
249
+ "load_args": false,
250
+ "load_data_args": false,
251
+ "packing": false,
252
+ "packing_length": null,
253
+ "packing_num_proc": 1,
254
+ "lazy_tokenize": false,
255
+ "use_hf": true,
256
+ "ignore_args_error": false,
257
+ "use_swift_lora": false,
258
+ "freeze_parameters": [],
259
+ "freeze_parameters_regex": null,
260
+ "freeze_parameters_ratio": 0.0,
261
+ "trainable_parameters": [],
262
+ "trainable_parameters_regex": null,
263
+ "freeze_llm": false,
264
+ "freeze_vit": true,
265
+ "freeze_aligner": false,
266
+ "target_modules": [
267
+ "all-linear"
268
+ ],
269
+ "target_regex": null,
270
+ "target_parameters": null,
271
+ "modules_to_save": [],
272
+ "lora_rank": 8,
273
+ "lora_alpha": 32,
274
+ "lora_dropout": 0.05,
275
+ "lora_bias": "none",
276
+ "lora_dtype": null,
277
+ "lorap_lr_ratio": 16.0,
278
+ "use_rslora": true,
279
+ "use_dora": false,
280
+ "lora_ga_batch_size": 2,
281
+ "lora_ga_iters": 2,
282
+ "lora_ga_max_length": 1024,
283
+ "lora_ga_direction": "ArB2r",
284
+ "lora_ga_scale": "stable",
285
+ "lora_ga_stable_gamma": 16,
286
+ "init_weights": true,
287
+ "fourier_n_frequency": 2000,
288
+ "fourier_scaling": 300.0,
289
+ "boft_block_size": 4,
290
+ "boft_block_num": 0,
291
+ "boft_n_butterfly_factor": 1,
292
+ "boft_dropout": 0.0,
293
+ "vera_rank": 256,
294
+ "vera_projection_prng_key": 0,
295
+ "vera_dropout": 0.0,
296
+ "vera_d_initial": 0.1,
297
+ "adapter_act": "gelu",
298
+ "adapter_length": 128,
299
+ "adalora_target_r": 8,
300
+ "adalora_init_r": 12,
301
+ "adalora_tinit": 0,
302
+ "adalora_tfinal": 0,
303
+ "adalora_deltaT": 1,
304
+ "adalora_beta1": 0.85,
305
+ "adalora_beta2": 0.85,
306
+ "adalora_orth_reg_weight": 0.5,
307
+ "llamapro_num_new_blocks": 4,
308
+ "llamapro_num_groups": null,
309
+ "reft_layer_key": null,
310
+ "reft_layers": null,
311
+ "reft_rank": 4,
312
+ "reft_intervention_type": "LoreftIntervention",
313
+ "reft_args": null,
314
+ "swanlab_token": null,
315
+ "swanlab_project": "ms-swift",
316
+ "swanlab_workspace": null,
317
+ "swanlab_exp_name": null,
318
+ "swanlab_notification_method": null,
319
+ "swanlab_webhook_url": null,
320
+ "swanlab_secret": null,
321
+ "swanlab_sender_email": null,
322
+ "swanlab_receiver_email": null,
323
+ "swanlab_smtp_server": null,
324
+ "swanlab_smtp_port": null,
325
+ "swanlab_email_language": "zh",
326
+ "swanlab_mode": "cloud",
327
+ "add_version": true,
328
+ "create_checkpoint_symlink": true,
329
+ "zero_hpz_partition_size": null,
330
+ "deepspeed_autotp_size": null,
331
+ "swift_version": "4.1.0",
332
+ "ckpt_dir": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-aligner/v0-20260424-001639/checkpoint-453",
333
+ "rank": -1,
334
+ "global_world_size": 1,
335
+ "local_world_size": 1,
336
+ "model_suffix": "Qwen3.5-4B",
337
+ "model_info": "ModelInfo(model_type='qwen3_5', model_dir='/home/kiol/.cache/huggingface/hub/models--Qwen--Qwen3.5-4B/snapshots/851bf6e806efd8d0a36b00ddf55e13ccb7b8cd0a', torch_dtype=torch.bfloat16, max_model_len=262144, quant_method='bnb', quant_bits=4, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
338
+ "model_meta": "ModelMeta(model_type='qwen3_5', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen3.5-0.8B', hf_model_id='Qwen/Qwen3.5-0.8B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-2B', hf_model_id='Qwen/Qwen3.5-2B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-4B', hf_model_id='Qwen/Qwen3.5-4B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-9B', hf_model_id='Qwen/Qwen3.5-9B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-27B', hf_model_id='Qwen/Qwen3.5-27B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-27B-FP8', hf_model_id='Qwen/Qwen3.5-27B-FP8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-0.8B-Base', hf_model_id='Qwen/Qwen3.5-0.8B-Base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-2B-Base', hf_model_id='Qwen/Qwen3.5-2B-Base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-4B-Base', hf_model_id='Qwen/Qwen3.5-4B-Base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3.5-9B-Base', hf_model_id='Qwen/Qwen3.5-9B-Base', model_path=None, ms_revision=None, hf_revision=None)], template='qwen3_5', ignore_patterns=None, requires=None, tags=[])], loader=<class 'swift.model.models.qwen.Qwen3_5Loader'>, template='qwen3_5', model_arch=MultiModelKeys(arch_name='qwen2_vl', embedding=None, module_list=None, lm_head=None, q_proj=None, k_proj=None, v_proj=None, o_proj=None, attention=None, mlp=None, down_proj=None, qkv_proj=None, qk_proj=None, qa_proj=None, qb_proj=None, kv_proj=None, kva_proj=None, kvb_proj=None, language_model=['model.language_model', 'lm_head'], aligner=['model.visual.merger'], vision_tower=['model.visual'], generator=[]), mcore_model_type=None, architectures=['Qwen3_5ForConditionalGeneration'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=5.0.0.dev', 'qwen_vl_utils>=0.0.14', 'decord'], tags=[])",
339
+ "model_dir": "/home/kiol/.cache/huggingface/hub/models--Qwen--Qwen3.5-4B/snapshots/851bf6e806efd8d0a36b00ddf55e13ccb7b8cd0a",
340
+ "template_meta": "QwenTemplateMeta(template_type='qwen3_5', prefix=[], prompt=['<|im_start|>user\\n{{QUERY}}<|im_end|>\\n<|im_start|>assistant\\n'], chat_sep=['<|im_end|>\\n'], suffix=['<|im_end|>\\n'], template_cls=<class 'swift.template.templates.qwen.Qwen3_5Template'>, system_prefix=['<|im_start|>system\\n{{SYSTEM}}<|im_end|>\\n'], default_system=None, auto_add_bos=False, stop_words=['<|endoftext|>'], agent_template='qwen3_5', is_thinking=True, thinking_prefix='<think>\\n', non_thinking_prefix='<think>\\n\\n</think>\\n\\n', history_thinking_prefix='')",
341
+ "_val_dataset_exists": true,
342
+ "hub": "<class 'swift.hub.hub.HFHub'>",
343
+ "evaluation_strategy": "steps",
344
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332', per_device_train_batch_size=1, num_train_epochs=3.0, max_steps=1359, learning_rate=0.0001, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_steps=0.05, optim=<OptimizerNames.ADAMW_TORCH_FUSED: 'adamw_torch_fused'>, optim_args=None, weight_decay=0.1, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, optim_target_modules=None, gradient_accumulation_steps=32, average_tokens_across_devices=None, max_grad_norm=1.0, label_smoothing_factor=0.0, bf16=True, fp16=False, bf16_full_eval=False, fp16_full_eval=False, tf32=None, gradient_checkpointing=True, gradient_checkpointing_kwargs=None, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, use_liger_kernel=False, liger_kernel_config=None, use_cache=False, neftune_noise_alpha=None, torch_empty_cache_steps=None, auto_find_batch_size=False, logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_steps=5, logging_first_step=True, log_on_each_node=True, logging_nan_inf_filter=True, include_num_input_tokens_seen=None, log_level='passive', log_level_replica='warning', disable_tqdm=False, report_to=['tensorboard'], run_name='/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332', project='huggingface', trackio_space_id='trackio', eval_strategy=<IntervalStrategy.STEPS: 'steps'>, eval_steps=200, eval_delay=0, per_device_eval_batch_size=1, prediction_loss_only=False, eval_on_start=False, eval_do_concat_batches=True, eval_use_gather_object=False, eval_accumulation_steps=None, include_for_metrics=[], batch_eval_metrics=False, save_only_model=False, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=200, save_on_each_node=False, save_total_limit=2, enable_jit_checkpoint=False, push_to_hub=False, hub_token=None, hub_private_repo=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_always_push=False, hub_revision=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=True, restore_callback_states_from_checkpoint=False, full_determinism=False, seed=23, data_seed=42, use_cpu=False, accelerator_config=AcceleratorConfig(split_batches=False, dispatch_batches=False, even_batches=True, use_seedable_sampler=True, non_blocking=False, gradient_accumulation_kwargs=None, use_configured_state=False), parallelism_config=None, dataloader_drop_last=False, dataloader_num_workers=4, dataloader_pin_memory=True, dataloader_persistent_workers=False, dataloader_prefetch_factor=2, remove_unused_columns=False, label_names=None, train_sampling_strategy='random', length_column_name='length', ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, ddp_backend=None, ddp_timeout=18000000, fsdp=[], fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, deepspeed=None, debug=[], skip_memory_metrics=True, do_train=False, do_eval=True, do_predict=False, resume_from_checkpoint='/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-aligner/v0-20260424-001639/checkpoint-453', warmup_ratio=0.05, logging_dir='/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332/runs', local_rank=-1, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=False, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, safe_serialization=True, max_shard_size='5GB', check_model=True, acc_strategy='token', train_dataloader_shuffle=True, group_by_length=True, max_epochs=None, aligner_lr=None, vit_lr=None, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=True, optimizer='lorap', loss_type=None, eval_metric=None, callbacks=[], early_stop_interval=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, tuner_type='lora', use_galore=False, galore_target_modules=None, galore_rank=128, galore_update_proj_gap=50, galore_scale=1.0, galore_proj_type='std', galore_optim_per_parameter=False, galore_with_embedding=False, galore_quantization=False, galore_proj_quant=False, galore_proj_bits=4, galore_proj_group_size=256, galore_cos_threshold=0.4, galore_gamma_proj=2, galore_queue_size=5, lisa_activated_layers=0, lisa_step_interval=20, use_flash_ckpt=False)"
345
+ }
labels.json ADDED
@@ -0,0 +1,408 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "supported_labels": [
3
+ "PLANT_01",
4
+ "PLANT_02",
5
+ "PLANT_03",
6
+ "PLANT_04",
7
+ "PLANT_05",
8
+ "PLANT_06",
9
+ "PLANT_07",
10
+ "PLANT_08",
11
+ "PLANT_09",
12
+ "PLANT_10",
13
+ "PLANT_11",
14
+ "PLANT_12",
15
+ "PLANT_13",
16
+ "PLANT_14",
17
+ "PLANT_15",
18
+ "PLANT_16",
19
+ "PLANT_17",
20
+ "PLANT_18",
21
+ "PLANT_19",
22
+ "PLANT_20",
23
+ "FUN_01",
24
+ "FUN_02",
25
+ "FUN_03",
26
+ "FUN_04",
27
+ "FUN_05",
28
+ "FUN_06",
29
+ "FUN_07",
30
+ "FUN_08",
31
+ "FUN_09",
32
+ "FUN_10",
33
+ "FUN_11",
34
+ "FUN_12",
35
+ "FUN_13",
36
+ "FUN_14",
37
+ "FUN_15",
38
+ "FUN_16",
39
+ "FUN_17",
40
+ "FUN_18",
41
+ "FUN_19",
42
+ "FUN_20",
43
+ "unknown"
44
+ ],
45
+ "training_supported_labels": [
46
+ "PLANT_01",
47
+ "PLANT_02",
48
+ "PLANT_03",
49
+ "PLANT_04",
50
+ "PLANT_05",
51
+ "PLANT_06",
52
+ "PLANT_07",
53
+ "PLANT_08",
54
+ "PLANT_09",
55
+ "PLANT_10",
56
+ "PLANT_11",
57
+ "PLANT_12",
58
+ "PLANT_13",
59
+ "PLANT_14",
60
+ "PLANT_15",
61
+ "PLANT_16",
62
+ "PLANT_17",
63
+ "PLANT_18",
64
+ "PLANT_19",
65
+ "PLANT_20",
66
+ "FUN_01",
67
+ "FUN_02",
68
+ "FUN_03",
69
+ "FUN_04",
70
+ "FUN_05",
71
+ "FUN_06",
72
+ "FUN_07",
73
+ "FUN_08",
74
+ "FUN_09",
75
+ "FUN_10",
76
+ "FUN_11",
77
+ "FUN_12",
78
+ "FUN_13",
79
+ "FUN_14",
80
+ "FUN_15",
81
+ "FUN_16",
82
+ "FUN_17",
83
+ "FUN_18",
84
+ "FUN_19",
85
+ "FUN_20",
86
+ "UNK_NON_TARGET_PLANT",
87
+ "UNK_NON_TARGET_FUNGUS",
88
+ "UNK_OTHER_OR_AMBIGUOUS"
89
+ ],
90
+ "collapse_to_public": {
91
+ "UNK_NON_TARGET_PLANT": "unknown",
92
+ "UNK_NON_TARGET_FUNGUS": "unknown",
93
+ "UNK_OTHER_OR_AMBIGUOUS": "unknown"
94
+ },
95
+ "by_id": {
96
+ "PLANT_01": {
97
+ "label_id": "PLANT_01",
98
+ "kingdom": "plants",
99
+ "scientific_name": "Pinus sylvestris",
100
+ "polish_name": "sosna zwyczajna",
101
+ "english_name": "Scots pine"
102
+ },
103
+ "PLANT_02": {
104
+ "label_id": "PLANT_02",
105
+ "kingdom": "plants",
106
+ "scientific_name": "Picea abies",
107
+ "polish_name": "swierk pospolity",
108
+ "english_name": "Norway spruce"
109
+ },
110
+ "PLANT_03": {
111
+ "label_id": "PLANT_03",
112
+ "kingdom": "plants",
113
+ "scientific_name": "Betula pendula",
114
+ "polish_name": "brzoza brodawkowata",
115
+ "english_name": "silver birch"
116
+ },
117
+ "PLANT_04": {
118
+ "label_id": "PLANT_04",
119
+ "kingdom": "plants",
120
+ "scientific_name": "Betula pubescens",
121
+ "polish_name": "brzoza omszona",
122
+ "english_name": "downy birch"
123
+ },
124
+ "PLANT_05": {
125
+ "label_id": "PLANT_05",
126
+ "kingdom": "plants",
127
+ "scientific_name": "Quercus robur",
128
+ "polish_name": "dab szypulkowy",
129
+ "english_name": "pedunculate oak"
130
+ },
131
+ "PLANT_06": {
132
+ "label_id": "PLANT_06",
133
+ "kingdom": "plants",
134
+ "scientific_name": "Quercus petraea",
135
+ "polish_name": "dab bezszypulkowy",
136
+ "english_name": "sessile oak"
137
+ },
138
+ "PLANT_07": {
139
+ "label_id": "PLANT_07",
140
+ "kingdom": "plants",
141
+ "scientific_name": "Fagus sylvatica",
142
+ "polish_name": "buk zwyczajny",
143
+ "english_name": "European beech"
144
+ },
145
+ "PLANT_08": {
146
+ "label_id": "PLANT_08",
147
+ "kingdom": "plants",
148
+ "scientific_name": "Alnus glutinosa",
149
+ "polish_name": "olsza czarna",
150
+ "english_name": "black alder"
151
+ },
152
+ "PLANT_09": {
153
+ "label_id": "PLANT_09",
154
+ "kingdom": "plants",
155
+ "scientific_name": "Populus tremula",
156
+ "polish_name": "osika",
157
+ "english_name": "European aspen"
158
+ },
159
+ "PLANT_10": {
160
+ "label_id": "PLANT_10",
161
+ "kingdom": "plants",
162
+ "scientific_name": "Acer platanoides",
163
+ "polish_name": "klon zwyczajny",
164
+ "english_name": "Norway maple"
165
+ },
166
+ "PLANT_11": {
167
+ "label_id": "PLANT_11",
168
+ "kingdom": "plants",
169
+ "scientific_name": "Sorbus aucuparia",
170
+ "polish_name": "jarzab pospolity",
171
+ "english_name": "rowan"
172
+ },
173
+ "PLANT_12": {
174
+ "label_id": "PLANT_12",
175
+ "kingdom": "plants",
176
+ "scientific_name": "Corylus avellana",
177
+ "polish_name": "leszczyna pospolita",
178
+ "english_name": "common hazel"
179
+ },
180
+ "PLANT_13": {
181
+ "label_id": "PLANT_13",
182
+ "kingdom": "plants",
183
+ "scientific_name": "Vaccinium myrtillus",
184
+ "polish_name": "borowka czarna",
185
+ "english_name": "bilberry"
186
+ },
187
+ "PLANT_14": {
188
+ "label_id": "PLANT_14",
189
+ "kingdom": "plants",
190
+ "scientific_name": "Vaccinium vitis-idaea",
191
+ "polish_name": "borowka brusznica",
192
+ "english_name": "lingonberry"
193
+ },
194
+ "PLANT_15": {
195
+ "label_id": "PLANT_15",
196
+ "kingdom": "plants",
197
+ "scientific_name": "Calluna vulgaris",
198
+ "polish_name": "wrzos zwyczajny",
199
+ "english_name": "heather"
200
+ },
201
+ "PLANT_16": {
202
+ "label_id": "PLANT_16",
203
+ "kingdom": "plants",
204
+ "scientific_name": "Convallaria majalis",
205
+ "polish_name": "konwalia majowa",
206
+ "english_name": "lily of the valley"
207
+ },
208
+ "PLANT_17": {
209
+ "label_id": "PLANT_17",
210
+ "kingdom": "plants",
211
+ "scientific_name": "Oxalis acetosella",
212
+ "polish_name": "szczawik zajeczy",
213
+ "english_name": "wood sorrel"
214
+ },
215
+ "PLANT_18": {
216
+ "label_id": "PLANT_18",
217
+ "kingdom": "plants",
218
+ "scientific_name": "Anemone nemorosa",
219
+ "polish_name": "zawilec gajowy",
220
+ "english_name": "wood anemone"
221
+ },
222
+ "PLANT_19": {
223
+ "label_id": "PLANT_19",
224
+ "kingdom": "plants",
225
+ "scientific_name": "Maianthemum bifolium",
226
+ "polish_name": "konwalijka dwulistna",
227
+ "english_name": "false lily of the valley"
228
+ },
229
+ "PLANT_20": {
230
+ "label_id": "PLANT_20",
231
+ "kingdom": "plants",
232
+ "scientific_name": "Pteridium aquilinum",
233
+ "polish_name": "orlica pospolita",
234
+ "english_name": "bracken"
235
+ },
236
+ "FUN_01": {
237
+ "label_id": "FUN_01",
238
+ "kingdom": "fungi",
239
+ "scientific_name": "Boletus edulis",
240
+ "polish_name": "borowik szlachetny",
241
+ "english_name": "king bolete"
242
+ },
243
+ "FUN_02": {
244
+ "label_id": "FUN_02",
245
+ "kingdom": "fungi",
246
+ "scientific_name": "Leccinum scabrum",
247
+ "polish_name": "kozlarz babka",
248
+ "english_name": "birch bolete"
249
+ },
250
+ "FUN_03": {
251
+ "label_id": "FUN_03",
252
+ "kingdom": "fungi",
253
+ "scientific_name": "Leccinum aurantiacum",
254
+ "polish_name": "kozlarz czerwony",
255
+ "english_name": "red-capped scaber stalk"
256
+ },
257
+ "FUN_04": {
258
+ "label_id": "FUN_04",
259
+ "kingdom": "fungi",
260
+ "scientific_name": "Cantharellus cibarius",
261
+ "polish_name": "pieprznik jadalny",
262
+ "english_name": "chanterelle"
263
+ },
264
+ "FUN_05": {
265
+ "label_id": "FUN_05",
266
+ "kingdom": "fungi",
267
+ "scientific_name": "Suillus luteus",
268
+ "polish_name": "maslak zwyczajny",
269
+ "english_name": "slippery jack"
270
+ },
271
+ "FUN_06": {
272
+ "label_id": "FUN_06",
273
+ "kingdom": "fungi",
274
+ "scientific_name": "Suillus bovinus",
275
+ "polish_name": "maslak sitarz",
276
+ "english_name": "Jersey cow mushroom"
277
+ },
278
+ "FUN_07": {
279
+ "label_id": "FUN_07",
280
+ "kingdom": "fungi",
281
+ "scientific_name": "Lactarius deliciosus",
282
+ "polish_name": "mleczaj rydz",
283
+ "english_name": "saffron milk cap"
284
+ },
285
+ "FUN_08": {
286
+ "label_id": "FUN_08",
287
+ "kingdom": "fungi",
288
+ "scientific_name": "Lactarius deterrimus",
289
+ "polish_name": "mleczaj swierkowy",
290
+ "english_name": "spruce milk cap"
291
+ },
292
+ "FUN_09": {
293
+ "label_id": "FUN_09",
294
+ "kingdom": "fungi",
295
+ "scientific_name": "Russula cyanoxantha",
296
+ "polish_name": "golabek zielonawofioletowy",
297
+ "english_name": "charcoal burner"
298
+ },
299
+ "FUN_10": {
300
+ "label_id": "FUN_10",
301
+ "kingdom": "fungi",
302
+ "scientific_name": "Russula claroflava",
303
+ "polish_name": "golabek blotny",
304
+ "english_name": "yellow swamp russula"
305
+ },
306
+ "FUN_11": {
307
+ "label_id": "FUN_11",
308
+ "kingdom": "fungi",
309
+ "scientific_name": "Macrolepiota procera",
310
+ "polish_name": "czubajka kania",
311
+ "english_name": "parasol mushroom"
312
+ },
313
+ "FUN_12": {
314
+ "label_id": "FUN_12",
315
+ "kingdom": "fungi",
316
+ "scientific_name": "Amanita muscaria",
317
+ "polish_name": "muchomor czerwony",
318
+ "english_name": "fly agaric"
319
+ },
320
+ "FUN_13": {
321
+ "label_id": "FUN_13",
322
+ "kingdom": "fungi",
323
+ "scientific_name": "Amanita phalloides",
324
+ "polish_name": "muchomor sromotnikowy",
325
+ "english_name": "death cap"
326
+ },
327
+ "FUN_14": {
328
+ "label_id": "FUN_14",
329
+ "kingdom": "fungi",
330
+ "scientific_name": "Armillaria mellea",
331
+ "polish_name": "opienka miodowa",
332
+ "english_name": "honey fungus"
333
+ },
334
+ "FUN_15": {
335
+ "label_id": "FUN_15",
336
+ "kingdom": "fungi",
337
+ "scientific_name": "Coprinus comatus",
338
+ "polish_name": "czernidlak kolpakowaty",
339
+ "english_name": "shaggy ink cap"
340
+ },
341
+ "FUN_16": {
342
+ "label_id": "FUN_16",
343
+ "kingdom": "fungi",
344
+ "scientific_name": "Lycoperdon perlatum",
345
+ "polish_name": "purchawka chropowata",
346
+ "english_name": "common puffball"
347
+ },
348
+ "FUN_17": {
349
+ "label_id": "FUN_17",
350
+ "kingdom": "fungi",
351
+ "scientific_name": "Phallus impudicus",
352
+ "polish_name": "sromotnik bezwstydny",
353
+ "english_name": "common stinkhorn"
354
+ },
355
+ "FUN_18": {
356
+ "label_id": "FUN_18",
357
+ "kingdom": "fungi",
358
+ "scientific_name": "Fomitopsis pinicola",
359
+ "polish_name": "pniarek obrzezony",
360
+ "english_name": "red-belted bracket"
361
+ },
362
+ "FUN_19": {
363
+ "label_id": "FUN_19",
364
+ "kingdom": "fungi",
365
+ "scientific_name": "Trametes versicolor",
366
+ "polish_name": "wrosniak roznobarwny",
367
+ "english_name": "turkey tail"
368
+ },
369
+ "FUN_20": {
370
+ "label_id": "FUN_20",
371
+ "kingdom": "fungi",
372
+ "scientific_name": "Xerocomellus chrysenteron",
373
+ "polish_name": "podgrzybek czerwonawy",
374
+ "english_name": "red cracking bolete"
375
+ },
376
+ "unknown": {
377
+ "label_id": "unknown",
378
+ "kingdom": "unknown",
379
+ "scientific_name": "unknown",
380
+ "polish_name": "nieznane",
381
+ "english_name": "unknown"
382
+ },
383
+ "UNK_NON_TARGET_PLANT": {
384
+ "label_id": "UNK_NON_TARGET_PLANT",
385
+ "kingdom": "plants",
386
+ "scientific_name": "unknown_non_target_plant",
387
+ "polish_name": "nieznana roslina spoza taksonomii",
388
+ "english_name": "unknown non-target plant",
389
+ "public_label_id": "unknown"
390
+ },
391
+ "UNK_NON_TARGET_FUNGUS": {
392
+ "label_id": "UNK_NON_TARGET_FUNGUS",
393
+ "kingdom": "fungi",
394
+ "scientific_name": "unknown_non_target_fungus",
395
+ "polish_name": "nieznany grzyb spoza taksonomii",
396
+ "english_name": "unknown non-target fungus",
397
+ "public_label_id": "unknown"
398
+ },
399
+ "UNK_OTHER_OR_AMBIGUOUS": {
400
+ "label_id": "UNK_OTHER_OR_AMBIGUOUS",
401
+ "kingdom": "unknown",
402
+ "scientific_name": "unknown_other_or_ambiguous",
403
+ "polish_name": "inne lub niejednoznaczne",
404
+ "english_name": "other or ambiguous",
405
+ "public_label_id": "unknown"
406
+ }
407
+ }
408
+ }
species_manifest.csv ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ label_id,kingdom,scientific_name,polish_name,english_name
2
+ PLANT_01,plants,Pinus sylvestris,sosna zwyczajna,Scots pine
3
+ PLANT_02,plants,Picea abies,swierk pospolity,Norway spruce
4
+ PLANT_03,plants,Betula pendula,brzoza brodawkowata,silver birch
5
+ PLANT_04,plants,Betula pubescens,brzoza omszona,downy birch
6
+ PLANT_05,plants,Quercus robur,dab szypulkowy,pedunculate oak
7
+ PLANT_06,plants,Quercus petraea,dab bezszypulkowy,sessile oak
8
+ PLANT_07,plants,Fagus sylvatica,buk zwyczajny,European beech
9
+ PLANT_08,plants,Alnus glutinosa,olsza czarna,black alder
10
+ PLANT_09,plants,Populus tremula,osika,European aspen
11
+ PLANT_10,plants,Acer platanoides,klon zwyczajny,Norway maple
12
+ PLANT_11,plants,Sorbus aucuparia,jarzab pospolity,rowan
13
+ PLANT_12,plants,Corylus avellana,leszczyna pospolita,common hazel
14
+ PLANT_13,plants,Vaccinium myrtillus,borowka czarna,bilberry
15
+ PLANT_14,plants,Vaccinium vitis-idaea,borowka brusznica,lingonberry
16
+ PLANT_15,plants,Calluna vulgaris,wrzos zwyczajny,heather
17
+ PLANT_16,plants,Convallaria majalis,konwalia majowa,lily of the valley
18
+ PLANT_17,plants,Oxalis acetosella,szczawik zajeczy,wood sorrel
19
+ PLANT_18,plants,Anemone nemorosa,zawilec gajowy,wood anemone
20
+ PLANT_19,plants,Maianthemum bifolium,konwalijka dwulistna,false lily of the valley
21
+ PLANT_20,plants,Pteridium aquilinum,orlica pospolita,bracken
22
+ FUN_01,fungi,Boletus edulis,borowik szlachetny,king bolete
23
+ FUN_02,fungi,Leccinum scabrum,kozlarz babka,birch bolete
24
+ FUN_03,fungi,Leccinum aurantiacum,kozlarz czerwony,red-capped scaber stalk
25
+ FUN_04,fungi,Cantharellus cibarius,pieprznik jadalny,chanterelle
26
+ FUN_05,fungi,Suillus luteus,maslak zwyczajny,slippery jack
27
+ FUN_06,fungi,Suillus bovinus,maslak sitarz,Jersey cow mushroom
28
+ FUN_07,fungi,Lactarius deliciosus,mleczaj rydz,saffron milk cap
29
+ FUN_08,fungi,Lactarius deterrimus,mleczaj swierkowy,spruce milk cap
30
+ FUN_09,fungi,Russula cyanoxantha,golabek zielonawofioletowy,charcoal burner
31
+ FUN_10,fungi,Russula claroflava,golabek blotny,yellow swamp russula
32
+ FUN_11,fungi,Macrolepiota procera,czubajka kania,parasol mushroom
33
+ FUN_12,fungi,Amanita muscaria,muchomor czerwony,fly agaric
34
+ FUN_13,fungi,Amanita phalloides,muchomor sromotnikowy,death cap
35
+ FUN_14,fungi,Armillaria mellea,opienka miodowa,honey fungus
36
+ FUN_15,fungi,Coprinus comatus,czernidlak kolpakowaty,shaggy ink cap
37
+ FUN_16,fungi,Lycoperdon perlatum,purchawka chropowata,common puffball
38
+ FUN_17,fungi,Phallus impudicus,sromotnik bezwstydny,common stinkhorn
39
+ FUN_18,fungi,Fomitopsis pinicola,pniarek obrzezony,red-belted bracket
40
+ FUN_19,fungi,Trametes versicolor,wrosniak roznobarwny,turkey tail
41
+ FUN_20,fungi,Xerocomellus chrysenteron,podgrzybek czerwonawy,red cracking bolete
trainer_state.json ADDED
@@ -0,0 +1,2282 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 1359,
3
+ "best_metric": 0.05789753,
4
+ "best_model_checkpoint": "/home/kiol/runs-v2-full-clean-r1/qwen35-qlora-forest/v0-20260424-170332/checkpoint-1359",
5
+ "epoch": 3.0,
6
+ "eval_steps": 200,
7
+ "global_step": 1359,
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.002210708117443869,
14
+ "grad_norm": 2.3902769088745117,
15
+ "learning_rate": 2.173913043478261e-06,
16
+ "loss": 0.5411382913589478,
17
+ "step": 1,
18
+ "token_acc": 0.8289156626506025
19
+ },
20
+ {
21
+ "epoch": 0.011053540587219343,
22
+ "grad_norm": 2.6440868377685547,
23
+ "learning_rate": 1.0869565217391305e-05,
24
+ "loss": 0.5509852766990662,
25
+ "step": 5,
26
+ "token_acc": 0.8427947598253275
27
+ },
28
+ {
29
+ "epoch": 0.022107081174438686,
30
+ "grad_norm": 2.8517158031463623,
31
+ "learning_rate": 2.173913043478261e-05,
32
+ "loss": 0.5423528671264648,
33
+ "step": 10,
34
+ "token_acc": 0.8413485374318295
35
+ },
36
+ {
37
+ "epoch": 0.03316062176165803,
38
+ "grad_norm": 3.078794479370117,
39
+ "learning_rate": 3.260869565217392e-05,
40
+ "loss": 0.4979440689086914,
41
+ "step": 15,
42
+ "token_acc": 0.8463063956370848
43
+ },
44
+ {
45
+ "epoch": 0.04421416234887737,
46
+ "grad_norm": 3.6568212509155273,
47
+ "learning_rate": 4.347826086956522e-05,
48
+ "loss": 0.4140318393707275,
49
+ "step": 20,
50
+ "token_acc": 0.8631159780985564
51
+ },
52
+ {
53
+ "epoch": 0.055267702936096716,
54
+ "grad_norm": 2.4268248081207275,
55
+ "learning_rate": 4.9997331144187255e-05,
56
+ "loss": 0.292206335067749,
57
+ "step": 25,
58
+ "token_acc": 0.9111776447105788
59
+ },
60
+ {
61
+ "epoch": 0.06632124352331606,
62
+ "grad_norm": 3.583557605743408,
63
+ "learning_rate": 4.996731305997416e-05,
64
+ "loss": 0.2570572137832642,
65
+ "step": 30,
66
+ "token_acc": 0.9214214214214215
67
+ },
68
+ {
69
+ "epoch": 0.0773747841105354,
70
+ "grad_norm": 3.932553768157959,
71
+ "learning_rate": 4.990398100856367e-05,
72
+ "loss": 0.20562427043914794,
73
+ "step": 35,
74
+ "token_acc": 0.9317507418397626
75
+ },
76
+ {
77
+ "epoch": 0.08842832469775475,
78
+ "grad_norm": 2.240527391433716,
79
+ "learning_rate": 4.980741949411839e-05,
80
+ "loss": 0.21206333637237548,
81
+ "step": 40,
82
+ "token_acc": 0.932
83
+ },
84
+ {
85
+ "epoch": 0.09948186528497409,
86
+ "grad_norm": 2.624908447265625,
87
+ "learning_rate": 4.967775735898179e-05,
88
+ "loss": 0.21093401908874512,
89
+ "step": 45,
90
+ "token_acc": 0.9330024813895782
91
+ },
92
+ {
93
+ "epoch": 0.11053540587219343,
94
+ "grad_norm": 1.4830845594406128,
95
+ "learning_rate": 4.9515167611763434e-05,
96
+ "loss": 0.1953817367553711,
97
+ "step": 50,
98
+ "token_acc": 0.9338308457711443
99
+ },
100
+ {
101
+ "epoch": 0.12158894645941278,
102
+ "grad_norm": 2.913989305496216,
103
+ "learning_rate": 4.931986719649299e-05,
104
+ "loss": 0.16938740015029907,
105
+ "step": 55,
106
+ "token_acc": 0.9406234537357744
107
+ },
108
+ {
109
+ "epoch": 0.13264248704663212,
110
+ "grad_norm": 2.571464776992798,
111
+ "learning_rate": 4.909211670315114e-05,
112
+ "loss": 0.18647342920303345,
113
+ "step": 60,
114
+ "token_acc": 0.9361914257228315
115
+ },
116
+ {
117
+ "epoch": 0.14369602763385148,
118
+ "grad_norm": 2.3652737140655518,
119
+ "learning_rate": 4.8832220019963514e-05,
120
+ "loss": 0.16005475521087648,
121
+ "step": 65,
122
+ "token_acc": 0.9426719840478565
123
+ },
124
+ {
125
+ "epoch": 0.1547495682210708,
126
+ "grad_norm": 2.1498990058898926,
127
+ "learning_rate": 4.8540523927921616e-05,
128
+ "loss": 0.17245489358901978,
129
+ "step": 70,
130
+ "token_acc": 0.9344672336168084
131
+ },
132
+ {
133
+ "epoch": 0.16580310880829016,
134
+ "grad_norm": 2.2657954692840576,
135
+ "learning_rate": 4.821741763807186e-05,
136
+ "loss": 0.1709640145301819,
137
+ "step": 75,
138
+ "token_acc": 0.9368473396320238
139
+ },
140
+ {
141
+ "epoch": 0.1768566493955095,
142
+ "grad_norm": 2.3392601013183594,
143
+ "learning_rate": 4.786333227218995e-05,
144
+ "loss": 0.16215839385986328,
145
+ "step": 80,
146
+ "token_acc": 0.9388667992047713
147
+ },
148
+ {
149
+ "epoch": 0.18791018998272885,
150
+ "grad_norm": 3.259490489959717,
151
+ "learning_rate": 4.747874028753375e-05,
152
+ "loss": 0.16813175678253173,
153
+ "step": 85,
154
+ "token_acc": 0.9411177644710579
155
+ },
156
+ {
157
+ "epoch": 0.19896373056994818,
158
+ "grad_norm": 2.805907726287842,
159
+ "learning_rate": 4.706415484644195e-05,
160
+ "loss": 0.16006848812103272,
161
+ "step": 90,
162
+ "token_acc": 0.944666001994018
163
+ },
164
+ {
165
+ "epoch": 0.21001727115716753,
166
+ "grad_norm": 2.46066951751709,
167
+ "learning_rate": 4.662012913161997e-05,
168
+ "loss": 0.15026205778121948,
169
+ "step": 95,
170
+ "token_acc": 0.9417910447761194
171
+ },
172
+ {
173
+ "epoch": 0.22107081174438686,
174
+ "grad_norm": 2.5938236713409424,
175
+ "learning_rate": 4.6147255608026394e-05,
176
+ "loss": 0.15484832525253295,
177
+ "step": 100,
178
+ "token_acc": 0.9400099157164105
179
+ },
180
+ {
181
+ "epoch": 0.23212435233160622,
182
+ "grad_norm": 2.2375521659851074,
183
+ "learning_rate": 4.564616523234511e-05,
184
+ "loss": 0.16213221549987794,
185
+ "step": 105,
186
+ "token_acc": 0.9409704852426213
187
+ },
188
+ {
189
+ "epoch": 0.24317789291882555,
190
+ "grad_norm": 2.779315233230591,
191
+ "learning_rate": 4.511752661109768e-05,
192
+ "loss": 0.15692013502120972,
193
+ "step": 110,
194
+ "token_acc": 0.9458519622454049
195
+ },
196
+ {
197
+ "epoch": 0.2542314335060449,
198
+ "grad_norm": 2.144108295440674,
199
+ "learning_rate": 4.4562045108519565e-05,
200
+ "loss": 0.1405424118041992,
201
+ "step": 115,
202
+ "token_acc": 0.9512437810945273
203
+ },
204
+ {
205
+ "epoch": 0.26528497409326424,
206
+ "grad_norm": 2.1729953289031982,
207
+ "learning_rate": 4.398046190539025e-05,
208
+ "loss": 0.13323441743850709,
209
+ "step": 120,
210
+ "token_acc": 0.9441953163926258
211
+ },
212
+ {
213
+ "epoch": 0.2763385146804836,
214
+ "grad_norm": 2.5072147846221924,
215
+ "learning_rate": 4.3373553010073355e-05,
216
+ "loss": 0.1478448271751404,
217
+ "step": 125,
218
+ "token_acc": 0.9433681073025335
219
+ },
220
+ {
221
+ "epoch": 0.28739205526770295,
222
+ "grad_norm": 3.7264955043792725,
223
+ "learning_rate": 4.274212822308612e-05,
224
+ "loss": 0.13619284629821776,
225
+ "step": 130,
226
+ "token_acc": 0.9491778774289985
227
+ },
228
+ {
229
+ "epoch": 0.29844559585492225,
230
+ "grad_norm": 3.988476276397705,
231
+ "learning_rate": 4.208703005657999e-05,
232
+ "loss": 0.1341521382331848,
233
+ "step": 135,
234
+ "token_acc": 0.9512922465208747
235
+ },
236
+ {
237
+ "epoch": 0.3094991364421416,
238
+ "grad_norm": 3.7429981231689453,
239
+ "learning_rate": 4.140913261017382e-05,
240
+ "loss": 0.13466038703918456,
241
+ "step": 140,
242
+ "token_acc": 0.9512074913750616
243
+ },
244
+ {
245
+ "epoch": 0.32055267702936097,
246
+ "grad_norm": 2.5042636394500732,
247
+ "learning_rate": 4.070934040463998e-05,
248
+ "loss": 0.13430129289627074,
249
+ "step": 145,
250
+ "token_acc": 0.9491525423728814
251
+ },
252
+ {
253
+ "epoch": 0.3316062176165803,
254
+ "grad_norm": 3.5047497749328613,
255
+ "learning_rate": 3.998858717499931e-05,
256
+ "loss": 0.1267208695411682,
257
+ "step": 150,
258
+ "token_acc": 0.9575636545182227
259
+ },
260
+ {
261
+ "epoch": 0.3426597582037997,
262
+ "grad_norm": 3.953479528427124,
263
+ "learning_rate": 3.924783462463541e-05,
264
+ "loss": 0.11603262424468994,
265
+ "step": 155,
266
+ "token_acc": 0.9582297364495276
267
+ },
268
+ {
269
+ "epoch": 0.353713298791019,
270
+ "grad_norm": 4.1243791580200195,
271
+ "learning_rate": 3.848807114209074e-05,
272
+ "loss": 0.1214677095413208,
273
+ "step": 160,
274
+ "token_acc": 0.9600591715976331
275
+ },
276
+ {
277
+ "epoch": 0.36476683937823834,
278
+ "grad_norm": 4.218201160430908,
279
+ "learning_rate": 3.7710310482256526e-05,
280
+ "loss": 0.12736471891403198,
281
+ "step": 165,
282
+ "token_acc": 0.9536390827517448
283
+ },
284
+ {
285
+ "epoch": 0.3758203799654577,
286
+ "grad_norm": 5.623844623565674,
287
+ "learning_rate": 3.691559041371631e-05,
288
+ "loss": 0.11791330575942993,
289
+ "step": 170,
290
+ "token_acc": 0.9556109725685785
291
+ },
292
+ {
293
+ "epoch": 0.38687392055267705,
294
+ "grad_norm": 2.831035852432251,
295
+ "learning_rate": 3.6104971334047956e-05,
296
+ "loss": 0.10760444402694702,
297
+ "step": 175,
298
+ "token_acc": 0.9595461272816971
299
+ },
300
+ {
301
+ "epoch": 0.39792746113989635,
302
+ "grad_norm": 3.888122797012329,
303
+ "learning_rate": 3.527953485493168e-05,
304
+ "loss": 0.11927105188369751,
305
+ "step": 180,
306
+ "token_acc": 0.956737941322725
307
+ },
308
+ {
309
+ "epoch": 0.4089810017271157,
310
+ "grad_norm": 3.259575128555298,
311
+ "learning_rate": 3.444038235895212e-05,
312
+ "loss": 0.12609773874282837,
313
+ "step": 185,
314
+ "token_acc": 0.955
315
+ },
316
+ {
317
+ "epoch": 0.42003454231433507,
318
+ "grad_norm": 4.910781383514404,
319
+ "learning_rate": 3.358863353001987e-05,
320
+ "loss": 0.12669839859008789,
321
+ "step": 190,
322
+ "token_acc": 0.9573412698412699
323
+ },
324
+ {
325
+ "epoch": 0.4310880829015544,
326
+ "grad_norm": 2.9438295364379883,
327
+ "learning_rate": 3.272542485937369e-05,
328
+ "loss": 0.11617603302001953,
329
+ "step": 195,
330
+ "token_acc": 0.958958958958959
331
+ },
332
+ {
333
+ "epoch": 0.4421416234887737,
334
+ "grad_norm": 5.1633195877075195,
335
+ "learning_rate": 3.185190812915646e-05,
336
+ "loss": 0.10707591772079468,
337
+ "step": 200,
338
+ "token_acc": 0.959479739869935
339
+ },
340
+ {
341
+ "epoch": 0.4421416234887737,
342
+ "eval_loss": 0.11586810648441315,
343
+ "eval_runtime": 1992.0002,
344
+ "eval_samples_per_second": 1.178,
345
+ "eval_steps_per_second": 1.178,
346
+ "eval_token_acc": 0.9600378980137381,
347
+ "step": 200
348
+ },
349
+ {
350
+ "epoch": 0.4531951640759931,
351
+ "grad_norm": 4.371393203735352,
352
+ "learning_rate": 3.096924887558855e-05,
353
+ "loss": 0.11527643203735352,
354
+ "step": 205,
355
+ "token_acc": 0.9623202776400595
356
+ },
357
+ {
358
+ "epoch": 0.46424870466321244,
359
+ "grad_norm": 5.050312519073486,
360
+ "learning_rate": 3.007862483378906e-05,
361
+ "loss": 0.12530215978622436,
362
+ "step": 210,
363
+ "token_acc": 0.9602583209140586
364
+ },
365
+ {
366
+ "epoch": 0.4753022452504318,
367
+ "grad_norm": 2.541715145111084,
368
+ "learning_rate": 2.9181224366319947e-05,
369
+ "loss": 0.10488080978393555,
370
+ "step": 215,
371
+ "token_acc": 0.9631474103585658
372
+ },
373
+ {
374
+ "epoch": 0.4863557858376511,
375
+ "grad_norm": 5.020773410797119,
376
+ "learning_rate": 2.827824487755007e-05,
377
+ "loss": 0.13031703233718872,
378
+ "step": 220,
379
+ "token_acc": 0.9523809523809523
380
+ },
381
+ {
382
+ "epoch": 0.49740932642487046,
383
+ "grad_norm": 5.226169586181641,
384
+ "learning_rate": 2.7370891215954568e-05,
385
+ "loss": 0.13065972328186035,
386
+ "step": 225,
387
+ "token_acc": 0.9522150323544052
388
+ },
389
+ {
390
+ "epoch": 0.5084628670120898,
391
+ "grad_norm": 3.837796211242676,
392
+ "learning_rate": 2.646037406648165e-05,
393
+ "loss": 0.11477760076522828,
394
+ "step": 230,
395
+ "token_acc": 0.9575848303393214
396
+ },
397
+ {
398
+ "epoch": 0.5195164075993092,
399
+ "grad_norm": 3.0600061416625977,
400
+ "learning_rate": 2.5547908335131704e-05,
401
+ "loss": 0.1025763750076294,
402
+ "step": 235,
403
+ "token_acc": 0.9629446640316206
404
+ },
405
+ {
406
+ "epoch": 0.5305699481865285,
407
+ "grad_norm": 4.423151016235352,
408
+ "learning_rate": 2.4634711527904272e-05,
409
+ "loss": 0.11243470907211303,
410
+ "step": 240,
411
+ "token_acc": 0.9575212393803099
412
+ },
413
+ {
414
+ "epoch": 0.5416234887737479,
415
+ "grad_norm": 3.2377634048461914,
416
+ "learning_rate": 2.3722002126275824e-05,
417
+ "loss": 0.11171900033950806,
418
+ "step": 245,
419
+ "token_acc": 0.9635182408795602
420
+ },
421
+ {
422
+ "epoch": 0.5526770293609672,
423
+ "grad_norm": 3.354611396789551,
424
+ "learning_rate": 2.281099796137594e-05,
425
+ "loss": 0.10156643390655518,
426
+ "step": 250,
427
+ "token_acc": 0.965965965965966
428
+ },
429
+ {
430
+ "epoch": 0.5637305699481865,
431
+ "grad_norm": 4.366595268249512,
432
+ "learning_rate": 2.19029145890313e-05,
433
+ "loss": 0.09984329342842102,
434
+ "step": 255,
435
+ "token_acc": 0.9646061814556331
436
+ },
437
+ {
438
+ "epoch": 0.5747841105354059,
439
+ "grad_norm": 4.832893371582031,
440
+ "learning_rate": 2.0998963667845535e-05,
441
+ "loss": 0.08920307159423828,
442
+ "step": 260,
443
+ "token_acc": 0.9660847880299251
444
+ },
445
+ {
446
+ "epoch": 0.5858376511226252,
447
+ "grad_norm": 10.402474403381348,
448
+ "learning_rate": 2.0100351342479216e-05,
449
+ "loss": 0.10913920402526855,
450
+ "step": 265,
451
+ "token_acc": 0.9623389494549058
452
+ },
453
+ {
454
+ "epoch": 0.5968911917098445,
455
+ "grad_norm": 5.610108852386475,
456
+ "learning_rate": 1.9208276634287143e-05,
457
+ "loss": 0.11966934204101562,
458
+ "step": 270,
459
+ "token_acc": 0.9542060726729716
460
+ },
461
+ {
462
+ "epoch": 0.6079447322970639,
463
+ "grad_norm": 5.160745143890381,
464
+ "learning_rate": 1.832392984146018e-05,
465
+ "loss": 0.12343072891235352,
466
+ "step": 275,
467
+ "token_acc": 0.9561752988047809
468
+ },
469
+ {
470
+ "epoch": 0.6189982728842832,
471
+ "grad_norm": 4.599337100982666,
472
+ "learning_rate": 1.7448490950806552e-05,
473
+ "loss": 0.09566409587860107,
474
+ "step": 280,
475
+ "token_acc": 0.9677579365079365
476
+ },
477
+ {
478
+ "epoch": 0.6300518134715026,
479
+ "grad_norm": 4.375885486602783,
480
+ "learning_rate": 1.6583128063291576e-05,
481
+ "loss": 0.12313523292541503,
482
+ "step": 285,
483
+ "token_acc": 0.9544328875681031
484
+ },
485
+ {
486
+ "epoch": 0.6411053540587219,
487
+ "grad_norm": 6.371300220489502,
488
+ "learning_rate": 1.572899583543671e-05,
489
+ "loss": 0.10101137161254883,
490
+ "step": 290,
491
+ "token_acc": 0.96250616674889
492
+ },
493
+ {
494
+ "epoch": 0.6521588946459412,
495
+ "grad_norm": 4.738733291625977,
496
+ "learning_rate": 1.488723393865766e-05,
497
+ "loss": 0.10386931896209717,
498
+ "step": 295,
499
+ "token_acc": 0.963681592039801
500
+ },
501
+ {
502
+ "epoch": 0.6632124352331606,
503
+ "grad_norm": 6.841185569763184,
504
+ "learning_rate": 1.4058965538597033e-05,
505
+ "loss": 0.1303364634513855,
506
+ "step": 300,
507
+ "token_acc": 0.9556772908366534
508
+ },
509
+ {
510
+ "epoch": 0.67426597582038,
511
+ "grad_norm": 14.118032455444336,
512
+ "learning_rate": 1.3245295796480789e-05,
513
+ "loss": 0.12113407850265503,
514
+ "step": 305,
515
+ "token_acc": 0.9616342800199302
516
+ },
517
+ {
518
+ "epoch": 0.6853195164075994,
519
+ "grad_norm": 3.676441192626953,
520
+ "learning_rate": 1.2447310394498019e-05,
521
+ "loss": 0.09884743690490723,
522
+ "step": 310,
523
+ "token_acc": 0.9665518937530743
524
+ },
525
+ {
526
+ "epoch": 0.6963730569948187,
527
+ "grad_norm": 5.252943992614746,
528
+ "learning_rate": 1.1666074087171627e-05,
529
+ "loss": 0.1010090470314026,
530
+ "step": 315,
531
+ "token_acc": 0.9651741293532339
532
+ },
533
+ {
534
+ "epoch": 0.707426597582038,
535
+ "grad_norm": 4.592480182647705,
536
+ "learning_rate": 1.0902629280652931e-05,
537
+ "loss": 0.10196793079376221,
538
+ "step": 320,
539
+ "token_acc": 0.9611166500498505
540
+ },
541
+ {
542
+ "epoch": 0.7184801381692574,
543
+ "grad_norm": 6.885665416717529,
544
+ "learning_rate": 1.0157994641835736e-05,
545
+ "loss": 0.11955760717391968,
546
+ "step": 325,
547
+ "token_acc": 0.95773247140726
548
+ },
549
+ {
550
+ "epoch": 0.7295336787564767,
551
+ "grad_norm": 5.868223667144775,
552
+ "learning_rate": 9.433163739145773e-06,
553
+ "loss": 0.09359505772590637,
554
+ "step": 330,
555
+ "token_acc": 0.9641434262948207
556
+ },
557
+ {
558
+ "epoch": 0.740587219343696,
559
+ "grad_norm": 6.838636875152588,
560
+ "learning_rate": 8.729103716819112e-06,
561
+ "loss": 0.09510601162910462,
562
+ "step": 335,
563
+ "token_acc": 0.9697870232788509
564
+ },
565
+ {
566
+ "epoch": 0.7516407599309154,
567
+ "grad_norm": 5.597217559814453,
568
+ "learning_rate": 8.046754004438429e-06,
569
+ "loss": 0.0947553813457489,
570
+ "step": 340,
571
+ "token_acc": 0.9647992067426872
572
+ },
573
+ {
574
+ "epoch": 0.7626943005181347,
575
+ "grad_norm": 4.449385643005371,
576
+ "learning_rate": 7.387025063449082e-06,
577
+ "loss": 0.11492215394973755,
578
+ "step": 345,
579
+ "token_acc": 0.9548834903321765
580
+ },
581
+ {
582
+ "epoch": 0.7737478411053541,
583
+ "grad_norm": 5.535495758056641,
584
+ "learning_rate": 6.750797172327442e-06,
585
+ "loss": 0.10709909200668336,
586
+ "step": 350,
587
+ "token_acc": 0.9646590343454455
588
+ },
589
+ {
590
+ "epoch": 0.7848013816925734,
591
+ "grad_norm": 3.9693377017974854,
592
+ "learning_rate": 6.138919252022435e-06,
593
+ "loss": 0.09377566576004029,
594
+ "step": 355,
595
+ "token_acc": 0.963220675944334
596
+ },
597
+ {
598
+ "epoch": 0.7958549222797927,
599
+ "grad_norm": 4.245842456817627,
600
+ "learning_rate": 5.5522077332375436e-06,
601
+ "loss": 0.11761529445648193,
602
+ "step": 360,
603
+ "token_acc": 0.9603371343579573
604
+ },
605
+ {
606
+ "epoch": 0.8069084628670121,
607
+ "grad_norm": 5.86805534362793,
608
+ "learning_rate": 4.99144546706469e-06,
609
+ "loss": 0.11018631458282471,
610
+ "step": 365,
611
+ "token_acc": 0.9619000494804553
612
+ },
613
+ {
614
+ "epoch": 0.8179620034542314,
615
+ "grad_norm": 3.946427583694458,
616
+ "learning_rate": 4.457380680423434e-06,
617
+ "loss": 0.09712615013122558,
618
+ "step": 370,
619
+ "token_acc": 0.9691695673794132
620
+ },
621
+ {
622
+ "epoch": 0.8290155440414507,
623
+ "grad_norm": 3.8146259784698486,
624
+ "learning_rate": 3.950725977699396e-06,
625
+ "loss": 0.10309855937957764,
626
+ "step": 375,
627
+ "token_acc": 0.9674837418709354
628
+ },
629
+ {
630
+ "epoch": 0.8400690846286701,
631
+ "grad_norm": 6.905686855316162,
632
+ "learning_rate": 3.4721573899138743e-06,
633
+ "loss": 0.12204140424728394,
634
+ "step": 380,
635
+ "token_acc": 0.9587064676616915
636
+ },
637
+ {
638
+ "epoch": 0.8511226252158894,
639
+ "grad_norm": 5.100115776062012,
640
+ "learning_rate": 3.0223134726934472e-06,
641
+ "loss": 0.10104206800460816,
642
+ "step": 385,
643
+ "token_acc": 0.9645885286783042
644
+ },
645
+ {
646
+ "epoch": 0.8621761658031089,
647
+ "grad_norm": 6.091329574584961,
648
+ "learning_rate": 2.6017944542431393e-06,
649
+ "loss": 0.09742544889450074,
650
+ "step": 390,
651
+ "token_acc": 0.9626307922272048
652
+ },
653
+ {
654
+ "epoch": 0.8732297063903282,
655
+ "grad_norm": 5.460122585296631,
656
+ "learning_rate": 2.2111614344599683e-06,
657
+ "loss": 0.10580202341079711,
658
+ "step": 395,
659
+ "token_acc": 0.9581673306772909
660
+ },
661
+ {
662
+ "epoch": 0.8842832469775475,
663
+ "grad_norm": 3.7578561305999756,
664
+ "learning_rate": 1.8509356362554963e-06,
665
+ "loss": 0.09627346396446228,
666
+ "step": 400,
667
+ "token_acc": 0.9677898909811695
668
+ },
669
+ {
670
+ "epoch": 0.8842832469775475,
671
+ "eval_loss": 0.10459936410188675,
672
+ "eval_runtime": 2136.0319,
673
+ "eval_samples_per_second": 1.099,
674
+ "eval_steps_per_second": 1.099,
675
+ "eval_token_acc": 0.9643014245592664,
676
+ "step": 400
677
+ },
678
+ {
679
+ "epoch": 0.8953367875647669,
680
+ "grad_norm": 4.357903480529785,
681
+ "learning_rate": 1.5215977100864392e-06,
682
+ "loss": 0.11347759962081909,
683
+ "step": 405,
684
+ "token_acc": 0.9596814335490294
685
+ },
686
+ {
687
+ "epoch": 0.9063903281519862,
688
+ "grad_norm": 2.9337828159332275,
689
+ "learning_rate": 1.2235870926211619e-06,
690
+ "loss": 0.11455904245376587,
691
+ "step": 410,
692
+ "token_acc": 0.9613095238095238
693
+ },
694
+ {
695
+ "epoch": 0.9174438687392056,
696
+ "grad_norm": 2.6067817211151123,
697
+ "learning_rate": 9.573014203979242e-07,
698
+ "loss": 0.0897371768951416,
699
+ "step": 415,
700
+ "token_acc": 0.9702970297029703
701
+ },
702
+ {
703
+ "epoch": 0.9284974093264249,
704
+ "grad_norm": 4.809932708740234,
705
+ "learning_rate": 7.230959992571368e-07,
706
+ "loss": 0.10621033906936646,
707
+ "step": 420,
708
+ "token_acc": 0.9666001994017946
709
+ },
710
+ {
711
+ "epoch": 0.9395509499136442,
712
+ "grad_norm": 6.890130996704102,
713
+ "learning_rate": 5.212833302556258e-07,
714
+ "loss": 0.10602649450302123,
715
+ "step": 425,
716
+ "token_acc": 0.9638076351016361
717
+ },
718
+ {
719
+ "epoch": 0.9506044905008636,
720
+ "grad_norm": 4.578585147857666,
721
+ "learning_rate": 3.521326926954532e-07,
722
+ "loss": 0.08783534765243531,
723
+ "step": 430,
724
+ "token_acc": 0.9670822942643391
725
+ },
726
+ {
727
+ "epoch": 0.9616580310880829,
728
+ "grad_norm": 4.462143898010254,
729
+ "learning_rate": 2.158697848236607e-07,
730
+ "loss": 0.08878173232078553,
731
+ "step": 435,
732
+ "token_acc": 0.9683950617283951
733
+ },
734
+ {
735
+ "epoch": 0.9727115716753022,
736
+ "grad_norm": 3.5091371536254883,
737
+ "learning_rate": 1.1267642268238121e-07,
738
+ "loss": 0.11377729177474975,
739
+ "step": 440,
740
+ "token_acc": 0.9592647789369101
741
+ },
742
+ {
743
+ "epoch": 0.9837651122625216,
744
+ "grad_norm": 5.187692165374756,
745
+ "learning_rate": 4.26902975110749e-08,
746
+ "loss": 0.08824495673179626,
747
+ "step": 445,
748
+ "token_acc": 0.9681750372948782
749
+ },
750
+ {
751
+ "epoch": 0.9948186528497409,
752
+ "grad_norm": 4.813872337341309,
753
+ "learning_rate": 6.004792024680295e-09,
754
+ "loss": 0.12380951642990112,
755
+ "step": 450,
756
+ "token_acc": 0.9557213930348258
757
+ },
758
+ {
759
+ "epoch": 1.0,
760
+ "eval_loss": 0.10443862527608871,
761
+ "eval_runtime": 2140.8222,
762
+ "eval_samples_per_second": 1.096,
763
+ "eval_steps_per_second": 1.096,
764
+ "eval_token_acc": 0.9642337495347342,
765
+ "step": 453
766
+ },
767
+ {
768
+ "epoch": 1.0044214162348877,
769
+ "grad_norm": 2.683554172515869,
770
+ "learning_rate": 2.9411764705882355e-06,
771
+ "loss": 0.08228109031915665,
772
+ "step": 455,
773
+ "token_acc": 0.9726708074534162
774
+ },
775
+ {
776
+ "epoch": 1.015474956822107,
777
+ "grad_norm": 3.1228041648864746,
778
+ "learning_rate": 1.0294117647058824e-05,
779
+ "loss": 0.08381627202033996,
780
+ "step": 460,
781
+ "token_acc": 0.9706905116741182
782
+ },
783
+ {
784
+ "epoch": 1.0265284974093265,
785
+ "grad_norm": 4.306940078735352,
786
+ "learning_rate": 1.7647058823529414e-05,
787
+ "loss": 0.09802506566047668,
788
+ "step": 465,
789
+ "token_acc": 0.9675810473815462
790
+ },
791
+ {
792
+ "epoch": 1.0375820379965457,
793
+ "grad_norm": 3.2758865356445312,
794
+ "learning_rate": 2.5e-05,
795
+ "loss": 0.08836065530776978,
796
+ "step": 470,
797
+ "token_acc": 0.9721531576330183
798
+ },
799
+ {
800
+ "epoch": 1.048635578583765,
801
+ "grad_norm": 5.335510730743408,
802
+ "learning_rate": 3.235294117647059e-05,
803
+ "loss": 0.09924425482749939,
804
+ "step": 475,
805
+ "token_acc": 0.963220675944334
806
+ },
807
+ {
808
+ "epoch": 1.0596891191709845,
809
+ "grad_norm": 6.677077770233154,
810
+ "learning_rate": 3.970588235294117e-05,
811
+ "loss": 0.09869781732559205,
812
+ "step": 480,
813
+ "token_acc": 0.9626307922272048
814
+ },
815
+ {
816
+ "epoch": 1.0707426597582037,
817
+ "grad_norm": 7.235245227813721,
818
+ "learning_rate": 4.705882352941177e-05,
819
+ "loss": 0.09256232976913452,
820
+ "step": 485,
821
+ "token_acc": 0.967869500741473
822
+ },
823
+ {
824
+ "epoch": 1.0817962003454231,
825
+ "grad_norm": 2.617497682571411,
826
+ "learning_rate": 5.441176470588235e-05,
827
+ "loss": 0.11165302991867065,
828
+ "step": 490,
829
+ "token_acc": 0.9609804902451226
830
+ },
831
+ {
832
+ "epoch": 1.0928497409326425,
833
+ "grad_norm": 4.969885349273682,
834
+ "learning_rate": 6.176470588235295e-05,
835
+ "loss": 0.10181330442428589,
836
+ "step": 495,
837
+ "token_acc": 0.9626865671641791
838
+ },
839
+ {
840
+ "epoch": 1.103903281519862,
841
+ "grad_norm": 3.402235746383667,
842
+ "learning_rate": 6.911764705882354e-05,
843
+ "loss": 0.09162335991859435,
844
+ "step": 500,
845
+ "token_acc": 0.9699062654168722
846
+ },
847
+ {
848
+ "epoch": 1.1149568221070811,
849
+ "grad_norm": 3.815495729446411,
850
+ "learning_rate": 7.647058823529411e-05,
851
+ "loss": 0.11382390260696411,
852
+ "step": 505,
853
+ "token_acc": 0.954228855721393
854
+ },
855
+ {
856
+ "epoch": 1.1260103626943005,
857
+ "grad_norm": 4.899413108825684,
858
+ "learning_rate": 8.382352941176471e-05,
859
+ "loss": 0.1054335355758667,
860
+ "step": 510,
861
+ "token_acc": 0.9592850049652433
862
+ },
863
+ {
864
+ "epoch": 1.1370639032815197,
865
+ "grad_norm": 4.382949352264404,
866
+ "learning_rate": 9.11764705882353e-05,
867
+ "loss": 0.0767556071281433,
868
+ "step": 515,
869
+ "token_acc": 0.9732540861812778
870
+ },
871
+ {
872
+ "epoch": 1.1481174438687392,
873
+ "grad_norm": 4.010786533355713,
874
+ "learning_rate": 9.852941176470589e-05,
875
+ "loss": 0.0905315101146698,
876
+ "step": 520,
877
+ "token_acc": 0.9681274900398407
878
+ },
879
+ {
880
+ "epoch": 1.1591709844559586,
881
+ "grad_norm": 8.248323440551758,
882
+ "learning_rate": 9.99976313340166e-05,
883
+ "loss": 0.14047706127166748,
884
+ "step": 525,
885
+ "token_acc": 0.9527127924340468
886
+ },
887
+ {
888
+ "epoch": 1.170224525043178,
889
+ "grad_norm": 6.752266883850098,
890
+ "learning_rate": 9.998800901308916e-05,
891
+ "loss": 0.1028751015663147,
892
+ "step": 530,
893
+ "token_acc": 0.9620947630922694
894
+ },
895
+ {
896
+ "epoch": 1.1812780656303972,
897
+ "grad_norm": 4.091604709625244,
898
+ "learning_rate": 9.997098641899562e-05,
899
+ "loss": 0.1167901635169983,
900
+ "step": 535,
901
+ "token_acc": 0.9626679940268791
902
+ },
903
+ {
904
+ "epoch": 1.1923316062176166,
905
+ "grad_norm": 3.13606858253479,
906
+ "learning_rate": 9.994656607177722e-05,
907
+ "loss": 0.11653541326522827,
908
+ "step": 540,
909
+ "token_acc": 0.955050505050505
910
+ },
911
+ {
912
+ "epoch": 1.203385146804836,
913
+ "grad_norm": 4.286221027374268,
914
+ "learning_rate": 9.991475158664578e-05,
915
+ "loss": 0.09516905546188355,
916
+ "step": 545,
917
+ "token_acc": 0.968031968031968
918
+ },
919
+ {
920
+ "epoch": 1.2144386873920552,
921
+ "grad_norm": 5.616212368011475,
922
+ "learning_rate": 9.987554767344845e-05,
923
+ "loss": 0.12145495414733887,
924
+ "step": 550,
925
+ "token_acc": 0.9621890547263682
926
+ },
927
+ {
928
+ "epoch": 1.2254922279792746,
929
+ "grad_norm": 5.72205114364624,
930
+ "learning_rate": 9.982896013597038e-05,
931
+ "loss": 0.12007032632827759,
932
+ "step": 555,
933
+ "token_acc": 0.9582089552238806
934
+ },
935
+ {
936
+ "epoch": 1.236545768566494,
937
+ "grad_norm": 3.8363678455352783,
938
+ "learning_rate": 9.977499587107569e-05,
939
+ "loss": 0.08600590825080871,
940
+ "step": 560,
941
+ "token_acc": 0.967196819085487
942
+ },
943
+ {
944
+ "epoch": 1.2475993091537134,
945
+ "grad_norm": 2.206713914871216,
946
+ "learning_rate": 9.971366286768629e-05,
947
+ "loss": 0.11412311792373657,
948
+ "step": 565,
949
+ "token_acc": 0.961824491819534
950
+ },
951
+ {
952
+ "epoch": 1.2586528497409326,
953
+ "grad_norm": 2.48056697845459,
954
+ "learning_rate": 9.964497020559926e-05,
955
+ "loss": 0.09548119902610779,
956
+ "step": 570,
957
+ "token_acc": 0.968222442899702
958
+ },
959
+ {
960
+ "epoch": 1.269706390328152,
961
+ "grad_norm": 4.197746276855469,
962
+ "learning_rate": 9.956892805414272e-05,
963
+ "loss": 0.11575849056243896,
964
+ "step": 575,
965
+ "token_acc": 0.9597215315763302
966
+ },
967
+ {
968
+ "epoch": 1.2807599309153712,
969
+ "grad_norm": 3.9247732162475586,
970
+ "learning_rate": 9.948554767067025e-05,
971
+ "loss": 0.09247745275497436,
972
+ "step": 580,
973
+ "token_acc": 0.9692001987083955
974
+ },
975
+ {
976
+ "epoch": 1.2918134715025906,
977
+ "grad_norm": 3.7010436058044434,
978
+ "learning_rate": 9.93948413988944e-05,
979
+ "loss": 0.11627188920974732,
980
+ "step": 585,
981
+ "token_acc": 0.9602780536246276
982
+ },
983
+ {
984
+ "epoch": 1.30286701208981,
985
+ "grad_norm": 6.0411858558654785,
986
+ "learning_rate": 9.92968226670593e-05,
987
+ "loss": 0.09203023314476014,
988
+ "step": 590,
989
+ "token_acc": 0.9705882352941176
990
+ },
991
+ {
992
+ "epoch": 1.3139205526770295,
993
+ "grad_norm": 4.776832103729248,
994
+ "learning_rate": 9.919150598595276e-05,
995
+ "loss": 0.07992117404937744,
996
+ "step": 595,
997
+ "token_acc": 0.9711155378486056
998
+ },
999
+ {
1000
+ "epoch": 1.3249740932642486,
1001
+ "grad_norm": 2.1442465782165527,
1002
+ "learning_rate": 9.907890694675803e-05,
1003
+ "loss": 0.08411768078804016,
1004
+ "step": 600,
1005
+ "token_acc": 0.971301335972291
1006
+ },
1007
+ {
1008
+ "epoch": 1.3249740932642486,
1009
+ "eval_loss": 0.10072976350784302,
1010
+ "eval_runtime": 2790.9554,
1011
+ "eval_samples_per_second": 0.841,
1012
+ "eval_steps_per_second": 0.841,
1013
+ "eval_token_acc": 0.9663655128074984,
1014
+ "step": 600
1015
+ },
1016
+ {
1017
+ "epoch": 1.336027633851468,
1018
+ "grad_norm": 3.444389820098877,
1019
+ "learning_rate": 9.89590422187457e-05,
1020
+ "loss": 0.0943886399269104,
1021
+ "step": 605,
1022
+ "token_acc": 0.969261279127417
1023
+ },
1024
+ {
1025
+ "epoch": 1.3470811744386875,
1026
+ "grad_norm": 3.3243448734283447,
1027
+ "learning_rate": 9.883192954680593e-05,
1028
+ "loss": 0.07429519295692444,
1029
+ "step": 610,
1030
+ "token_acc": 0.9723046488625123
1031
+ },
1032
+ {
1033
+ "epoch": 1.3581347150259067,
1034
+ "grad_norm": 3.897686243057251,
1035
+ "learning_rate": 9.869758774882154e-05,
1036
+ "loss": 0.10087257623672485,
1037
+ "step": 615,
1038
+ "token_acc": 0.9645
1039
+ },
1040
+ {
1041
+ "epoch": 1.369188255613126,
1042
+ "grad_norm": 3.0886733531951904,
1043
+ "learning_rate": 9.855603671288215e-05,
1044
+ "loss": 0.0957147240638733,
1045
+ "step": 620,
1046
+ "token_acc": 0.9651394422310757
1047
+ },
1048
+ {
1049
+ "epoch": 1.3802417962003455,
1050
+ "grad_norm": 5.4413604736328125,
1051
+ "learning_rate": 9.840729739433992e-05,
1052
+ "loss": 0.0882586419582367,
1053
+ "step": 625,
1054
+ "token_acc": 0.9705441837244134
1055
+ },
1056
+ {
1057
+ "epoch": 1.391295336787565,
1058
+ "grad_norm": 3.051844358444214,
1059
+ "learning_rate": 9.82513918127073e-05,
1060
+ "loss": 0.09511439800262451,
1061
+ "step": 630,
1062
+ "token_acc": 0.9663532904502722
1063
+ },
1064
+ {
1065
+ "epoch": 1.402348877374784,
1066
+ "grad_norm": 2.9811363220214844,
1067
+ "learning_rate": 9.808834304839729e-05,
1068
+ "loss": 0.10007621049880981,
1069
+ "step": 635,
1070
+ "token_acc": 0.9672943508424182
1071
+ },
1072
+ {
1073
+ "epoch": 1.4134024179620035,
1074
+ "grad_norm": 3.030879497528076,
1075
+ "learning_rate": 9.791817523930653e-05,
1076
+ "loss": 0.08152814507484436,
1077
+ "step": 640,
1078
+ "token_acc": 0.9720837487537388
1079
+ },
1080
+ {
1081
+ "epoch": 1.4244559585492227,
1082
+ "grad_norm": 2.544180154800415,
1083
+ "learning_rate": 9.774091357724196e-05,
1084
+ "loss": 0.07389838099479676,
1085
+ "step": 645,
1086
+ "token_acc": 0.9755854509217738
1087
+ },
1088
+ {
1089
+ "epoch": 1.435509499136442,
1090
+ "grad_norm": 5.591005802154541,
1091
+ "learning_rate": 9.755658430419132e-05,
1092
+ "loss": 0.09485760927200318,
1093
+ "step": 650,
1094
+ "token_acc": 0.9637357178340785
1095
+ },
1096
+ {
1097
+ "epoch": 1.4465630397236615,
1098
+ "grad_norm": 4.964202880859375,
1099
+ "learning_rate": 9.736521470843838e-05,
1100
+ "loss": 0.08160382509231567,
1101
+ "step": 655,
1102
+ "token_acc": 0.9705441837244134
1103
+ },
1104
+ {
1105
+ "epoch": 1.457616580310881,
1106
+ "grad_norm": 4.984673500061035,
1107
+ "learning_rate": 9.7166833120523e-05,
1108
+ "loss": 0.08802146315574647,
1109
+ "step": 660,
1110
+ "token_acc": 0.9641076769690927
1111
+ },
1112
+ {
1113
+ "epoch": 1.4686701208981001,
1114
+ "grad_norm": 2.584303140640259,
1115
+ "learning_rate": 9.696146890904722e-05,
1116
+ "loss": 0.09760611653327941,
1117
+ "step": 665,
1118
+ "token_acc": 0.9701343952215032
1119
+ },
1120
+ {
1121
+ "epoch": 1.4797236614853195,
1122
+ "grad_norm": 2.9796664714813232,
1123
+ "learning_rate": 9.674915247632739e-05,
1124
+ "loss": 0.09098277688026428,
1125
+ "step": 670,
1126
+ "token_acc": 0.9660678642714571
1127
+ },
1128
+ {
1129
+ "epoch": 1.490777202072539,
1130
+ "grad_norm": 10.275652885437012,
1131
+ "learning_rate": 9.652991525389337e-05,
1132
+ "loss": 0.08257744312286378,
1133
+ "step": 675,
1134
+ "token_acc": 0.9722084367245658
1135
+ },
1136
+ {
1137
+ "epoch": 1.5018307426597581,
1138
+ "grad_norm": 4.8155131340026855,
1139
+ "learning_rate": 9.630378969783547e-05,
1140
+ "loss": 0.07055800557136535,
1141
+ "step": 680,
1142
+ "token_acc": 0.974090682610862
1143
+ },
1144
+ {
1145
+ "epoch": 1.5128842832469775,
1146
+ "grad_norm": 3.6014211177825928,
1147
+ "learning_rate": 9.607080928399958e-05,
1148
+ "loss": 0.09370391964912414,
1149
+ "step": 685,
1150
+ "token_acc": 0.9658584858980702
1151
+ },
1152
+ {
1153
+ "epoch": 1.523937823834197,
1154
+ "grad_norm": 2.675119400024414,
1155
+ "learning_rate": 9.58310085030313e-05,
1156
+ "loss": 0.09670426845550537,
1157
+ "step": 690,
1158
+ "token_acc": 0.9641969169567379
1159
+ },
1160
+ {
1161
+ "epoch": 1.5349913644214164,
1162
+ "grad_norm": 3.324349880218506,
1163
+ "learning_rate": 9.558442285527e-05,
1164
+ "loss": 0.08441510200500488,
1165
+ "step": 695,
1166
+ "token_acc": 0.9696819085487077
1167
+ },
1168
+ {
1169
+ "epoch": 1.5460449050086356,
1170
+ "grad_norm": 3.4498190879821777,
1171
+ "learning_rate": 9.533108884549333e-05,
1172
+ "loss": 0.06717776656150817,
1173
+ "step": 700,
1174
+ "token_acc": 0.975597609561753
1175
+ },
1176
+ {
1177
+ "epoch": 1.557098445595855,
1178
+ "grad_norm": 2.959386110305786,
1179
+ "learning_rate": 9.50710439775129e-05,
1180
+ "loss": 0.08139981031417846,
1181
+ "step": 705,
1182
+ "token_acc": 0.9747023809523809
1183
+ },
1184
+ {
1185
+ "epoch": 1.5681519861830742,
1186
+ "grad_norm": 2.3681604862213135,
1187
+ "learning_rate": 9.480432674862232e-05,
1188
+ "loss": 0.07764554619789124,
1189
+ "step": 710,
1190
+ "token_acc": 0.9675810473815462
1191
+ },
1192
+ {
1193
+ "epoch": 1.5792055267702936,
1194
+ "grad_norm": 2.840590715408325,
1195
+ "learning_rate": 9.453097664389789e-05,
1196
+ "loss": 0.08232161402702332,
1197
+ "step": 715,
1198
+ "token_acc": 0.9701789264413518
1199
+ },
1200
+ {
1201
+ "epoch": 1.590259067357513,
1202
+ "grad_norm": 2.655217409133911,
1203
+ "learning_rate": 9.425103413035335e-05,
1204
+ "loss": 0.0968110740184784,
1205
+ "step": 720,
1206
+ "token_acc": 0.9689534301452178
1207
+ },
1208
+ {
1209
+ "epoch": 1.6013126079447324,
1210
+ "grad_norm": 2.4610400199890137,
1211
+ "learning_rate": 9.396454065094891e-05,
1212
+ "loss": 0.09739276766777039,
1213
+ "step": 725,
1214
+ "token_acc": 0.964729259811227
1215
+ },
1216
+ {
1217
+ "epoch": 1.6123661485319516,
1218
+ "grad_norm": 4.132114887237549,
1219
+ "learning_rate": 9.367153861845617e-05,
1220
+ "loss": 0.08105069994926453,
1221
+ "step": 730,
1222
+ "token_acc": 0.9716981132075472
1223
+ },
1224
+ {
1225
+ "epoch": 1.623419689119171,
1226
+ "grad_norm": 3.3622491359710693,
1227
+ "learning_rate": 9.337207140917919e-05,
1228
+ "loss": 0.09018557667732238,
1229
+ "step": 735,
1230
+ "token_acc": 0.9642324888226528
1231
+ },
1232
+ {
1233
+ "epoch": 1.6344732297063902,
1234
+ "grad_norm": 2.985978364944458,
1235
+ "learning_rate": 9.306618335653307e-05,
1236
+ "loss": 0.08649082779884339,
1237
+ "step": 740,
1238
+ "token_acc": 0.9683011391778108
1239
+ },
1240
+ {
1241
+ "epoch": 1.6455267702936096,
1242
+ "grad_norm": 3.509003162384033,
1243
+ "learning_rate": 9.275391974448076e-05,
1244
+ "loss": 0.0744367241859436,
1245
+ "step": 745,
1246
+ "token_acc": 0.9770687936191426
1247
+ },
1248
+ {
1249
+ "epoch": 1.656580310880829,
1250
+ "grad_norm": 2.937560796737671,
1251
+ "learning_rate": 9.243532680082915e-05,
1252
+ "loss": 0.07034647464752197,
1253
+ "step": 750,
1254
+ "token_acc": 0.9767211490837048
1255
+ },
1256
+ {
1257
+ "epoch": 1.6676338514680484,
1258
+ "grad_norm": 3.8314759731292725,
1259
+ "learning_rate": 9.211045169038554e-05,
1260
+ "loss": 0.07900274395942689,
1261
+ "step": 755,
1262
+ "token_acc": 0.9711729622266402
1263
+ },
1264
+ {
1265
+ "epoch": 1.6786873920552678,
1266
+ "grad_norm": 5.002687931060791,
1267
+ "learning_rate": 9.17793425079753e-05,
1268
+ "loss": 0.07675303220748901,
1269
+ "step": 760,
1270
+ "token_acc": 0.9689349112426036
1271
+ },
1272
+ {
1273
+ "epoch": 1.689740932642487,
1274
+ "grad_norm": 1.7825733423233032,
1275
+ "learning_rate": 9.144204827132175e-05,
1276
+ "loss": 0.08085300326347351,
1277
+ "step": 765,
1278
+ "token_acc": 0.9716981132075472
1279
+ },
1280
+ {
1281
+ "epoch": 1.7007944732297062,
1282
+ "grad_norm": 1.8953182697296143,
1283
+ "learning_rate": 9.10986189137897e-05,
1284
+ "loss": 0.07259147167205811,
1285
+ "step": 770,
1286
+ "token_acc": 0.9760479041916168
1287
+ },
1288
+ {
1289
+ "epoch": 1.7118480138169256,
1290
+ "grad_norm": 3.7877821922302246,
1291
+ "learning_rate": 9.074910527699313e-05,
1292
+ "loss": 0.08823164105415345,
1293
+ "step": 775,
1294
+ "token_acc": 0.972139303482587
1295
+ },
1296
+ {
1297
+ "epoch": 1.722901554404145,
1298
+ "grad_norm": 3.614501714706421,
1299
+ "learning_rate": 9.039355910326863e-05,
1300
+ "loss": 0.10905979871749878,
1301
+ "step": 780,
1302
+ "token_acc": 0.9652087475149106
1303
+ },
1304
+ {
1305
+ "epoch": 1.7339550949913645,
1306
+ "grad_norm": 3.0243847370147705,
1307
+ "learning_rate": 9.00320330280154e-05,
1308
+ "loss": 0.07965280413627625,
1309
+ "step": 785,
1310
+ "token_acc": 0.9723593287265548
1311
+ },
1312
+ {
1313
+ "epoch": 1.7450086355785839,
1314
+ "grad_norm": 4.025770664215088,
1315
+ "learning_rate": 8.966458057190301e-05,
1316
+ "loss": 0.07108275294303894,
1317
+ "step": 790,
1318
+ "token_acc": 0.9760956175298805
1319
+ },
1320
+ {
1321
+ "epoch": 1.756062176165803,
1322
+ "grad_norm": 2.9816761016845703,
1323
+ "learning_rate": 8.92912561329482e-05,
1324
+ "loss": 0.0776334285736084,
1325
+ "step": 795,
1326
+ "token_acc": 0.9706467661691542
1327
+ },
1328
+ {
1329
+ "epoch": 1.7671157167530225,
1330
+ "grad_norm": 3.0242762565612793,
1331
+ "learning_rate": 8.891211497846171e-05,
1332
+ "loss": 0.07837628722190856,
1333
+ "step": 800,
1334
+ "token_acc": 0.974155069582505
1335
+ },
1336
+ {
1337
+ "epoch": 1.7671157167530225,
1338
+ "eval_loss": 0.08094792068004608,
1339
+ "eval_runtime": 2286.4128,
1340
+ "eval_samples_per_second": 1.026,
1341
+ "eval_steps_per_second": 1.026,
1342
+ "eval_token_acc": 0.9727608026257909,
1343
+ "step": 800
1344
+ },
1345
+ {
1346
+ "epoch": 1.7781692573402417,
1347
+ "grad_norm": 3.771763324737549,
1348
+ "learning_rate": 8.852721323686648e-05,
1349
+ "loss": 0.08302398324012757,
1350
+ "step": 805,
1351
+ "token_acc": 0.9725411882176734
1352
+ },
1353
+ {
1354
+ "epoch": 1.789222797927461,
1355
+ "grad_norm": 3.6321651935577393,
1356
+ "learning_rate": 8.813660788938833e-05,
1357
+ "loss": 0.06937822699546814,
1358
+ "step": 810,
1359
+ "token_acc": 0.9772839506172839
1360
+ },
1361
+ {
1362
+ "epoch": 1.8002763385146805,
1363
+ "grad_norm": 6.7241644859313965,
1364
+ "learning_rate": 8.774035676162043e-05,
1365
+ "loss": 0.05159105062484741,
1366
+ "step": 815,
1367
+ "token_acc": 0.9829488465396189
1368
+ },
1369
+ {
1370
+ "epoch": 1.8113298791019,
1371
+ "grad_norm": 4.365586757659912,
1372
+ "learning_rate": 8.733851851496268e-05,
1373
+ "loss": 0.08399490118026734,
1374
+ "step": 820,
1375
+ "token_acc": 0.9721254355400697
1376
+ },
1377
+ {
1378
+ "epoch": 1.8223834196891193,
1379
+ "grad_norm": 3.0722031593322754,
1380
+ "learning_rate": 8.693115263793747e-05,
1381
+ "loss": 0.07215502858161926,
1382
+ "step": 825,
1383
+ "token_acc": 0.9740648379052369
1384
+ },
1385
+ {
1386
+ "epoch": 1.8334369602763385,
1387
+ "grad_norm": 4.519292831420898,
1388
+ "learning_rate": 8.651831943738296e-05,
1389
+ "loss": 0.06996339559555054,
1390
+ "step": 830,
1391
+ "token_acc": 0.9770459081836327
1392
+ },
1393
+ {
1394
+ "epoch": 1.8444905008635577,
1395
+ "grad_norm": 4.166755199432373,
1396
+ "learning_rate": 8.610008002952513e-05,
1397
+ "loss": 0.07142719030380248,
1398
+ "step": 835,
1399
+ "token_acc": 0.9767211490837048
1400
+ },
1401
+ {
1402
+ "epoch": 1.8555440414507771,
1403
+ "grad_norm": 4.686975002288818,
1404
+ "learning_rate": 8.567649633093016e-05,
1405
+ "loss": 0.06802060008049012,
1406
+ "step": 840,
1407
+ "token_acc": 0.9775
1408
+ },
1409
+ {
1410
+ "epoch": 1.8665975820379965,
1411
+ "grad_norm": 3.0628044605255127,
1412
+ "learning_rate": 8.524763104933816e-05,
1413
+ "loss": 0.06818159222602845,
1414
+ "step": 845,
1415
+ "token_acc": 0.973644952759821
1416
+ },
1417
+ {
1418
+ "epoch": 1.877651122625216,
1419
+ "grad_norm": 3.939176321029663,
1420
+ "learning_rate": 8.481354767437988e-05,
1421
+ "loss": 0.07347306013107299,
1422
+ "step": 850,
1423
+ "token_acc": 0.972568578553616
1424
+ },
1425
+ {
1426
+ "epoch": 1.8887046632124354,
1427
+ "grad_norm": 2.8834567070007324,
1428
+ "learning_rate": 8.437431046817769e-05,
1429
+ "loss": 0.06994418501853943,
1430
+ "step": 855,
1431
+ "token_acc": 0.971712158808933
1432
+ },
1433
+ {
1434
+ "epoch": 1.8997582037996545,
1435
+ "grad_norm": 3.4866409301757812,
1436
+ "learning_rate": 8.392998445583212e-05,
1437
+ "loss": 0.07565975189208984,
1438
+ "step": 860,
1439
+ "token_acc": 0.9760956175298805
1440
+ },
1441
+ {
1442
+ "epoch": 1.910811744386874,
1443
+ "grad_norm": 3.4372429847717285,
1444
+ "learning_rate": 8.348063541579545e-05,
1445
+ "loss": 0.07984944581985473,
1446
+ "step": 865,
1447
+ "token_acc": 0.9727452923686819
1448
+ },
1449
+ {
1450
+ "epoch": 1.9218652849740931,
1451
+ "grad_norm": 4.127252578735352,
1452
+ "learning_rate": 8.302632987013388e-05,
1453
+ "loss": 0.07774015665054321,
1454
+ "step": 870,
1455
+ "token_acc": 0.9744872436218109
1456
+ },
1457
+ {
1458
+ "epoch": 1.9329188255613126,
1459
+ "grad_norm": 3.960955858230591,
1460
+ "learning_rate": 8.256713507467941e-05,
1461
+ "loss": 0.08486457467079163,
1462
+ "step": 875,
1463
+ "token_acc": 0.9731743666169895
1464
+ },
1465
+ {
1466
+ "epoch": 1.943972366148532,
1467
+ "grad_norm": 3.087674617767334,
1468
+ "learning_rate": 8.210311900907339e-05,
1469
+ "loss": 0.07507517337799072,
1470
+ "step": 880,
1471
+ "token_acc": 0.9787023278850916
1472
+ },
1473
+ {
1474
+ "epoch": 1.9550259067357514,
1475
+ "grad_norm": 2.5197293758392334,
1476
+ "learning_rate": 8.163435036670261e-05,
1477
+ "loss": 0.08100587725639344,
1478
+ "step": 885,
1479
+ "token_acc": 0.9724724724724725
1480
+ },
1481
+ {
1482
+ "epoch": 1.9660794473229708,
1483
+ "grad_norm": 1.6736986637115479,
1484
+ "learning_rate": 8.116089854452995e-05,
1485
+ "loss": 0.07375568151473999,
1486
+ "step": 890,
1487
+ "token_acc": 0.9772727272727273
1488
+ },
1489
+ {
1490
+ "epoch": 1.97713298791019,
1491
+ "grad_norm": 3.322634220123291,
1492
+ "learning_rate": 8.068283363282074e-05,
1493
+ "loss": 0.07798144817352295,
1494
+ "step": 895,
1495
+ "token_acc": 0.9738400789733465
1496
+ },
1497
+ {
1498
+ "epoch": 1.9881865284974092,
1499
+ "grad_norm": 1.9556145668029785,
1500
+ "learning_rate": 8.020022640476654e-05,
1501
+ "loss": 0.06203848123550415,
1502
+ "step": 900,
1503
+ "token_acc": 0.9791666666666666
1504
+ },
1505
+ {
1506
+ "epoch": 1.9992400690846286,
1507
+ "grad_norm": 3.805701494216919,
1508
+ "learning_rate": 7.971314830600783e-05,
1509
+ "loss": 0.06745657324790955,
1510
+ "step": 905,
1511
+ "token_acc": 0.9751491053677932
1512
+ },
1513
+ {
1514
+ "epoch": 2.0088428324697754,
1515
+ "grad_norm": 2.904778003692627,
1516
+ "learning_rate": 7.922167144405706e-05,
1517
+ "loss": 0.06268702149391174,
1518
+ "step": 910,
1519
+ "token_acc": 0.9777777777777777
1520
+ },
1521
+ {
1522
+ "epoch": 2.0198963730569948,
1523
+ "grad_norm": 3.050584554672241,
1524
+ "learning_rate": 7.87258685776239e-05,
1525
+ "loss": 0.07978938817977906,
1526
+ "step": 915,
1527
+ "token_acc": 0.9681116093672147
1528
+ },
1529
+ {
1530
+ "epoch": 2.030949913644214,
1531
+ "grad_norm": 3.2201292514801025,
1532
+ "learning_rate": 7.822581310584388e-05,
1533
+ "loss": 0.07445316910743713,
1534
+ "step": 920,
1535
+ "token_acc": 0.9754509018036072
1536
+ },
1537
+ {
1538
+ "epoch": 2.0420034542314336,
1539
+ "grad_norm": 4.6090545654296875,
1540
+ "learning_rate": 7.772157905741231e-05,
1541
+ "loss": 0.06728174090385437,
1542
+ "step": 925,
1543
+ "token_acc": 0.977205153617443
1544
+ },
1545
+ {
1546
+ "epoch": 2.053056994818653,
1547
+ "grad_norm": 3.3497800827026367,
1548
+ "learning_rate": 7.721324107962506e-05,
1549
+ "loss": 0.06557589173316955,
1550
+ "step": 930,
1551
+ "token_acc": 0.9775784753363229
1552
+ },
1553
+ {
1554
+ "epoch": 2.064110535405872,
1555
+ "grad_norm": 2.1937358379364014,
1556
+ "learning_rate": 7.670087442732763e-05,
1557
+ "loss": 0.05688057541847229,
1558
+ "step": 935,
1559
+ "token_acc": 0.981028457314029
1560
+ },
1561
+ {
1562
+ "epoch": 2.0751640759930914,
1563
+ "grad_norm": 5.671027660369873,
1564
+ "learning_rate": 7.618455495177445e-05,
1565
+ "loss": 0.08629457950592041,
1566
+ "step": 940,
1567
+ "token_acc": 0.9695304695304695
1568
+ },
1569
+ {
1570
+ "epoch": 2.086217616580311,
1571
+ "grad_norm": 3.635037899017334,
1572
+ "learning_rate": 7.566435908939967e-05,
1573
+ "loss": 0.0566463053226471,
1574
+ "step": 945,
1575
+ "token_acc": 0.9820717131474104
1576
+ },
1577
+ {
1578
+ "epoch": 2.09727115716753,
1579
+ "grad_norm": 3.0799365043640137,
1580
+ "learning_rate": 7.514036385050147e-05,
1581
+ "loss": 0.06808796525001526,
1582
+ "step": 950,
1583
+ "token_acc": 0.9766052762568442
1584
+ },
1585
+ {
1586
+ "epoch": 2.1083246977547496,
1587
+ "grad_norm": 5.38563871383667,
1588
+ "learning_rate": 7.461264680784151e-05,
1589
+ "loss": 0.07369622588157654,
1590
+ "step": 955,
1591
+ "token_acc": 0.9737363726461843
1592
+ },
1593
+ {
1594
+ "epoch": 2.119378238341969,
1595
+ "grad_norm": 3.824101448059082,
1596
+ "learning_rate": 7.408128608516077e-05,
1597
+ "loss": 0.06465582847595215,
1598
+ "step": 960,
1599
+ "token_acc": 0.9786917740336968
1600
+ },
1601
+ {
1602
+ "epoch": 2.1304317789291884,
1603
+ "grad_norm": 3.410547971725464,
1604
+ "learning_rate": 7.354636034561418e-05,
1605
+ "loss": 0.051229971647262576,
1606
+ "step": 965,
1607
+ "token_acc": 0.9821428571428571
1608
+ },
1609
+ {
1610
+ "epoch": 2.1414853195164074,
1611
+ "grad_norm": 3.721679210662842,
1612
+ "learning_rate": 7.30079487801252e-05,
1613
+ "loss": 0.06965258717536926,
1614
+ "step": 970,
1615
+ "token_acc": 0.9755854509217738
1616
+ },
1617
+ {
1618
+ "epoch": 2.152538860103627,
1619
+ "grad_norm": 2.798208236694336,
1620
+ "learning_rate": 7.246613109566238e-05,
1621
+ "loss": 0.07870134711265564,
1622
+ "step": 975,
1623
+ "token_acc": 0.9751243781094527
1624
+ },
1625
+ {
1626
+ "epoch": 2.1635924006908462,
1627
+ "grad_norm": 1.8357700109481812,
1628
+ "learning_rate": 7.192098750343935e-05,
1629
+ "loss": 0.0715235412120819,
1630
+ "step": 980,
1631
+ "token_acc": 0.97675568743818
1632
+ },
1633
+ {
1634
+ "epoch": 2.1746459412780657,
1635
+ "grad_norm": 3.544813632965088,
1636
+ "learning_rate": 7.137259870704036e-05,
1637
+ "loss": 0.055529987812042235,
1638
+ "step": 985,
1639
+ "token_acc": 0.9841112214498511
1640
+ },
1641
+ {
1642
+ "epoch": 2.185699481865285,
1643
+ "grad_norm": 2.2707366943359375,
1644
+ "learning_rate": 7.082104589047285e-05,
1645
+ "loss": 0.05665128231048584,
1646
+ "step": 990,
1647
+ "token_acc": 0.9795102448775612
1648
+ },
1649
+ {
1650
+ "epoch": 2.1967530224525045,
1651
+ "grad_norm": 4.324965476989746,
1652
+ "learning_rate": 7.026641070614884e-05,
1653
+ "loss": 0.06112373471260071,
1654
+ "step": 995,
1655
+ "token_acc": 0.9775112443778111
1656
+ },
1657
+ {
1658
+ "epoch": 2.207806563039724,
1659
+ "grad_norm": 3.2108352184295654,
1660
+ "learning_rate": 6.970877526279702e-05,
1661
+ "loss": 0.061422485113143924,
1662
+ "step": 1000,
1663
+ "token_acc": 0.9781854238968766
1664
+ },
1665
+ {
1666
+ "epoch": 2.207806563039724,
1667
+ "eval_loss": 0.07163181900978088,
1668
+ "eval_runtime": 2173.3435,
1669
+ "eval_samples_per_second": 1.08,
1670
+ "eval_steps_per_second": 1.08,
1671
+ "eval_token_acc": 0.9747910533617569,
1672
+ "step": 1000
1673
+ },
1674
+ {
1675
+ "epoch": 2.218860103626943,
1676
+ "grad_norm": 4.137533187866211,
1677
+ "learning_rate": 6.914822211330742e-05,
1678
+ "loss": 0.06986818313598633,
1679
+ "step": 1005,
1680
+ "token_acc": 0.9751367478866235
1681
+ },
1682
+ {
1683
+ "epoch": 2.2299136442141623,
1684
+ "grad_norm": 1.7135125398635864,
1685
+ "learning_rate": 6.858483424251001e-05,
1686
+ "loss": 0.0670811414718628,
1687
+ "step": 1010,
1688
+ "token_acc": 0.9776341948310139
1689
+ },
1690
+ {
1691
+ "epoch": 2.2409671848013817,
1692
+ "grad_norm": 2.10587739944458,
1693
+ "learning_rate": 6.801869505488969e-05,
1694
+ "loss": 0.06850314140319824,
1695
+ "step": 1015,
1696
+ "token_acc": 0.9761312779711586
1697
+ },
1698
+ {
1699
+ "epoch": 2.252020725388601,
1700
+ "grad_norm": 3.5003225803375244,
1701
+ "learning_rate": 6.744988836223893e-05,
1702
+ "loss": 0.06779593229293823,
1703
+ "step": 1020,
1704
+ "token_acc": 0.9794692038057086
1705
+ },
1706
+ {
1707
+ "epoch": 2.2630742659758205,
1708
+ "grad_norm": 2.2275619506835938,
1709
+ "learning_rate": 6.687849837125027e-05,
1710
+ "loss": 0.05072577595710755,
1711
+ "step": 1025,
1712
+ "token_acc": 0.9836309523809523
1713
+ },
1714
+ {
1715
+ "epoch": 2.2741278065630395,
1716
+ "grad_norm": 2.6360952854156494,
1717
+ "learning_rate": 6.630460967105018e-05,
1718
+ "loss": 0.05415867567062378,
1719
+ "step": 1030,
1720
+ "token_acc": 0.9781746031746031
1721
+ },
1722
+ {
1723
+ "epoch": 2.285181347150259,
1724
+ "grad_norm": 1.4451478719711304,
1725
+ "learning_rate": 6.572830722067653e-05,
1726
+ "loss": 0.055239105224609376,
1727
+ "step": 1035,
1728
+ "token_acc": 0.981094527363184
1729
+ },
1730
+ {
1731
+ "epoch": 2.2962348877374783,
1732
+ "grad_norm": 1.9783443212509155,
1733
+ "learning_rate": 6.5149676336501e-05,
1734
+ "loss": 0.05858151912689209,
1735
+ "step": 1040,
1736
+ "token_acc": 0.9850224663005491
1737
+ },
1738
+ {
1739
+ "epoch": 2.3072884283246977,
1740
+ "grad_norm": 2.043297529220581,
1741
+ "learning_rate": 6.456880267959894e-05,
1742
+ "loss": 0.05577117800712585,
1743
+ "step": 1045,
1744
+ "token_acc": 0.9800697558545092
1745
+ },
1746
+ {
1747
+ "epoch": 2.318341968911917,
1748
+ "grad_norm": 4.235545635223389,
1749
+ "learning_rate": 6.39857722430679e-05,
1750
+ "loss": 0.05815597772598267,
1751
+ "step": 1050,
1752
+ "token_acc": 0.9801587301587301
1753
+ },
1754
+ {
1755
+ "epoch": 2.3293955094991365,
1756
+ "grad_norm": 5.21077823638916,
1757
+ "learning_rate": 6.340067133929719e-05,
1758
+ "loss": 0.054069459438323975,
1759
+ "step": 1055,
1760
+ "token_acc": 0.9801291604570294
1761
+ },
1762
+ {
1763
+ "epoch": 2.340449050086356,
1764
+ "grad_norm": 2.011702537536621,
1765
+ "learning_rate": 6.281358658719011e-05,
1766
+ "loss": 0.07169802188873291,
1767
+ "step": 1060,
1768
+ "token_acc": 0.9752107089737233
1769
+ },
1770
+ {
1771
+ "epoch": 2.351502590673575,
1772
+ "grad_norm": 2.3686680793762207,
1773
+ "learning_rate": 6.22246048993407e-05,
1774
+ "loss": 0.05615015029907226,
1775
+ "step": 1065,
1776
+ "token_acc": 0.9800299550673989
1777
+ },
1778
+ {
1779
+ "epoch": 2.3625561312607943,
1780
+ "grad_norm": 7.002840995788574,
1781
+ "learning_rate": 6.163381346916732e-05,
1782
+ "loss": 0.06114639043807983,
1783
+ "step": 1070,
1784
+ "token_acc": 0.9760239760239761
1785
+ },
1786
+ {
1787
+ "epoch": 2.3736096718480137,
1788
+ "grad_norm": 4.902541160583496,
1789
+ "learning_rate": 6.104129975800427e-05,
1790
+ "loss": 0.07762741446495056,
1791
+ "step": 1075,
1792
+ "token_acc": 0.9730807577268196
1793
+ },
1794
+ {
1795
+ "epoch": 2.384663212435233,
1796
+ "grad_norm": 4.661471843719482,
1797
+ "learning_rate": 6.0447151482153955e-05,
1798
+ "loss": 0.06509206891059875,
1799
+ "step": 1080,
1800
+ "token_acc": 0.9771144278606965
1801
+ },
1802
+ {
1803
+ "epoch": 2.3957167530224526,
1804
+ "grad_norm": 1.999237298965454,
1805
+ "learning_rate": 5.985145659990138e-05,
1806
+ "loss": 0.06380823254585266,
1807
+ "step": 1085,
1808
+ "token_acc": 0.9796626984126984
1809
+ },
1810
+ {
1811
+ "epoch": 2.406770293609672,
1812
+ "grad_norm": 2.382763385772705,
1813
+ "learning_rate": 5.925430329849264e-05,
1814
+ "loss": 0.05442737936973572,
1815
+ "step": 1090,
1816
+ "token_acc": 0.9821073558648111
1817
+ },
1818
+ {
1819
+ "epoch": 2.4178238341968914,
1820
+ "grad_norm": 1.8456308841705322,
1821
+ "learning_rate": 5.865577998107961e-05,
1822
+ "loss": 0.04835757613182068,
1823
+ "step": 1095,
1824
+ "token_acc": 0.9831013916500994
1825
+ },
1826
+ {
1827
+ "epoch": 2.4288773747841104,
1828
+ "grad_norm": 4.092327117919922,
1829
+ "learning_rate": 5.805597525363263e-05,
1830
+ "loss": 0.07175707817077637,
1831
+ "step": 1100,
1832
+ "token_acc": 0.977589641434263
1833
+ },
1834
+ {
1835
+ "epoch": 2.43993091537133,
1836
+ "grad_norm": 2.878070831298828,
1837
+ "learning_rate": 5.745497791182325e-05,
1838
+ "loss": 0.054905033111572264,
1839
+ "step": 1105,
1840
+ "token_acc": 0.9791459781529295
1841
+ },
1842
+ {
1843
+ "epoch": 2.450984455958549,
1844
+ "grad_norm": 3.2568767070770264,
1845
+ "learning_rate": 5.685287692787883e-05,
1846
+ "loss": 0.060244417190551756,
1847
+ "step": 1110,
1848
+ "token_acc": 0.9787549407114624
1849
+ },
1850
+ {
1851
+ "epoch": 2.4620379965457686,
1852
+ "grad_norm": 3.9391286373138428,
1853
+ "learning_rate": 5.6249761437410895e-05,
1854
+ "loss": 0.07208690047264099,
1855
+ "step": 1115,
1856
+ "token_acc": 0.9765234765234765
1857
+ },
1858
+ {
1859
+ "epoch": 2.473091537132988,
1860
+ "grad_norm": 1.7733020782470703,
1861
+ "learning_rate": 5.5645720726219584e-05,
1862
+ "loss": 0.05974746346473694,
1863
+ "step": 1120,
1864
+ "token_acc": 0.9806066633515664
1865
+ },
1866
+ {
1867
+ "epoch": 2.4841450777202074,
1868
+ "grad_norm": 4.430456161499023,
1869
+ "learning_rate": 5.504084421707555e-05,
1870
+ "loss": 0.0642861008644104,
1871
+ "step": 1125,
1872
+ "token_acc": 0.9781529294935452
1873
+ },
1874
+ {
1875
+ "epoch": 2.495198618307427,
1876
+ "grad_norm": 2.7552433013916016,
1877
+ "learning_rate": 5.443522145648181e-05,
1878
+ "loss": 0.06047917008399963,
1879
+ "step": 1130,
1880
+ "token_acc": 0.9830677290836654
1881
+ },
1882
+ {
1883
+ "epoch": 2.506252158894646,
1884
+ "grad_norm": 1.864843487739563,
1885
+ "learning_rate": 5.3828942101417136e-05,
1886
+ "loss": 0.044628658890724184,
1887
+ "step": 1135,
1888
+ "token_acc": 0.9856719367588933
1889
+ },
1890
+ {
1891
+ "epoch": 2.5173056994818652,
1892
+ "grad_norm": 4.15530252456665,
1893
+ "learning_rate": 5.322209590606323e-05,
1894
+ "loss": 0.0662376880645752,
1895
+ "step": 1140,
1896
+ "token_acc": 0.9765234765234765
1897
+ },
1898
+ {
1899
+ "epoch": 2.5283592400690846,
1900
+ "grad_norm": 3.1273319721221924,
1901
+ "learning_rate": 5.2614772708517324e-05,
1902
+ "loss": 0.06211344003677368,
1903
+ "step": 1145,
1904
+ "token_acc": 0.9807217004448838
1905
+ },
1906
+ {
1907
+ "epoch": 2.539412780656304,
1908
+ "grad_norm": 2.2132909297943115,
1909
+ "learning_rate": 5.200706241749257e-05,
1910
+ "loss": 0.05160966515541077,
1911
+ "step": 1150,
1912
+ "token_acc": 0.9830423940149626
1913
+ },
1914
+ {
1915
+ "epoch": 2.5504663212435235,
1916
+ "grad_norm": 2.4676833152770996,
1917
+ "learning_rate": 5.1399054999007756e-05,
1918
+ "loss": 0.05153646469116211,
1919
+ "step": 1155,
1920
+ "token_acc": 0.9795816733067729
1921
+ },
1922
+ {
1923
+ "epoch": 2.5615198618307424,
1924
+ "grad_norm": 2.5266482830047607,
1925
+ "learning_rate": 5.079084046306877e-05,
1926
+ "loss": 0.05694507360458374,
1927
+ "step": 1160,
1928
+ "token_acc": 0.9826474962816063
1929
+ },
1930
+ {
1931
+ "epoch": 2.572573402417962,
1932
+ "grad_norm": 3.3381104469299316,
1933
+ "learning_rate": 5.018250885034328e-05,
1934
+ "loss": 0.056800955533981325,
1935
+ "step": 1165,
1936
+ "token_acc": 0.9816377171215881
1937
+ },
1938
+ {
1939
+ "epoch": 2.5836269430051813,
1940
+ "grad_norm": 3.0276429653167725,
1941
+ "learning_rate": 4.957415021883121e-05,
1942
+ "loss": 0.061768895387649535,
1943
+ "step": 1170,
1944
+ "token_acc": 0.9804413239719157
1945
+ },
1946
+ {
1947
+ "epoch": 2.5946804835924007,
1948
+ "grad_norm": 1.4948302507400513,
1949
+ "learning_rate": 4.89658546305323e-05,
1950
+ "loss": 0.048909342288970946,
1951
+ "step": 1175,
1952
+ "token_acc": 0.9846306395637084
1953
+ },
1954
+ {
1955
+ "epoch": 2.60573402417962,
1956
+ "grad_norm": 2.0935652256011963,
1957
+ "learning_rate": 4.835771213811336e-05,
1958
+ "loss": 0.05250586867332459,
1959
+ "step": 1180,
1960
+ "token_acc": 0.9846534653465346
1961
+ },
1962
+ {
1963
+ "epoch": 2.6167875647668395,
1964
+ "grad_norm": 2.5400516986846924,
1965
+ "learning_rate": 4.774981277157673e-05,
1966
+ "loss": 0.05397605299949646,
1967
+ "step": 1185,
1968
+ "token_acc": 0.9816740960871718
1969
+ },
1970
+ {
1971
+ "epoch": 2.627841105354059,
1972
+ "grad_norm": 1.8448779582977295,
1973
+ "learning_rate": 4.714224652493212e-05,
1974
+ "loss": 0.0550678551197052,
1975
+ "step": 1190,
1976
+ "token_acc": 0.9800299550673989
1977
+ },
1978
+ {
1979
+ "epoch": 2.638894645941278,
1980
+ "grad_norm": 2.3702774047851562,
1981
+ "learning_rate": 4.6535103342873885e-05,
1982
+ "loss": 0.055988776683807376,
1983
+ "step": 1195,
1984
+ "token_acc": 0.9795511221945137
1985
+ },
1986
+ {
1987
+ "epoch": 2.6499481865284973,
1988
+ "grad_norm": 3.8877556324005127,
1989
+ "learning_rate": 4.592847310746549e-05,
1990
+ "loss": 0.054580336809158324,
1991
+ "step": 1200,
1992
+ "token_acc": 0.9802078179119248
1993
+ },
1994
+ {
1995
+ "epoch": 2.6499481865284973,
1996
+ "eval_loss": 0.06643825024366379,
1997
+ "eval_runtime": 2174.0597,
1998
+ "eval_samples_per_second": 1.08,
1999
+ "eval_steps_per_second": 1.08,
2000
+ "eval_token_acc": 0.9772273542449159,
2001
+ "step": 1200
2002
+ },
2003
+ {
2004
+ "epoch": 2.6610017271157167,
2005
+ "grad_norm": 4.951501846313477,
2006
+ "learning_rate": 4.5322445624833255e-05,
2007
+ "loss": 0.05614232420921326,
2008
+ "step": 1205,
2009
+ "token_acc": 0.979571499750872
2010
+ },
2011
+ {
2012
+ "epoch": 2.672055267702936,
2013
+ "grad_norm": 1.8844807147979736,
2014
+ "learning_rate": 4.471711061187144e-05,
2015
+ "loss": 0.05867302417755127,
2016
+ "step": 1210,
2017
+ "token_acc": 0.9790523690773068
2018
+ },
2019
+ {
2020
+ "epoch": 2.6831088082901555,
2021
+ "grad_norm": 2.765587329864502,
2022
+ "learning_rate": 4.411255768296038e-05,
2023
+ "loss": 0.05516909956932068,
2024
+ "step": 1215,
2025
+ "token_acc": 0.9800697558545092
2026
+ },
2027
+ {
2028
+ "epoch": 2.694162348877375,
2029
+ "grad_norm": 3.7630977630615234,
2030
+ "learning_rate": 4.3508876336699974e-05,
2031
+ "loss": 0.05011783838272095,
2032
+ "step": 1220,
2033
+ "token_acc": 0.981555333998006
2034
+ },
2035
+ {
2036
+ "epoch": 2.7052158894645943,
2037
+ "grad_norm": 3.2775371074676514,
2038
+ "learning_rate": 4.290615594266013e-05,
2039
+ "loss": 0.04247501492500305,
2040
+ "step": 1225,
2041
+ "token_acc": 0.9855
2042
+ },
2043
+ {
2044
+ "epoch": 2.7162694300518133,
2045
+ "grad_norm": 2.6794540882110596,
2046
+ "learning_rate": 4.230448572815053e-05,
2047
+ "loss": 0.04988014101982117,
2048
+ "step": 1230,
2049
+ "token_acc": 0.9826130153999006
2050
+ },
2051
+ {
2052
+ "epoch": 2.7273229706390327,
2053
+ "grad_norm": 2.045193910598755,
2054
+ "learning_rate": 4.170395476501119e-05,
2055
+ "loss": 0.04342162907123566,
2056
+ "step": 1235,
2057
+ "token_acc": 0.9841269841269841
2058
+ },
2059
+ {
2060
+ "epoch": 2.738376511226252,
2061
+ "grad_norm": 2.58183217048645,
2062
+ "learning_rate": 4.1104651956426296e-05,
2063
+ "loss": 0.04766501486301422,
2064
+ "step": 1240,
2065
+ "token_acc": 0.9831097863884749
2066
+ },
2067
+ {
2068
+ "epoch": 2.7494300518134716,
2069
+ "grad_norm": 3.195136785507202,
2070
+ "learning_rate": 4.050666602376287e-05,
2071
+ "loss": 0.05665205121040344,
2072
+ "step": 1245,
2073
+ "token_acc": 0.9766749379652605
2074
+ },
2075
+ {
2076
+ "epoch": 2.760483592400691,
2077
+ "grad_norm": 3.2019379138946533,
2078
+ "learning_rate": 3.991008549343626e-05,
2079
+ "loss": 0.07038918733596802,
2080
+ "step": 1250,
2081
+ "token_acc": 0.9775449101796407
2082
+ },
2083
+ {
2084
+ "epoch": 2.77153713298791,
2085
+ "grad_norm": 4.3877854347229,
2086
+ "learning_rate": 3.931499868380482e-05,
2087
+ "loss": 0.06642740964889526,
2088
+ "step": 1255,
2089
+ "token_acc": 0.9761904761904762
2090
+ },
2091
+ {
2092
+ "epoch": 2.78259067357513,
2093
+ "grad_norm": 1.3834000825881958,
2094
+ "learning_rate": 3.872149369209491e-05,
2095
+ "loss": 0.0616798460483551,
2096
+ "step": 1260,
2097
+ "token_acc": 0.979571499750872
2098
+ },
2099
+ {
2100
+ "epoch": 2.7936442141623488,
2101
+ "grad_norm": 4.06312370300293,
2102
+ "learning_rate": 3.8129658381359156e-05,
2103
+ "loss": 0.07107862830162048,
2104
+ "step": 1265,
2105
+ "token_acc": 0.9755244755244755
2106
+ },
2107
+ {
2108
+ "epoch": 2.804697754749568,
2109
+ "grad_norm": 3.0957045555114746,
2110
+ "learning_rate": 3.753958036746894e-05,
2111
+ "loss": 0.0476302444934845,
2112
+ "step": 1270,
2113
+ "token_acc": 0.9815645241654211
2114
+ },
2115
+ {
2116
+ "epoch": 2.8157512953367876,
2117
+ "grad_norm": 4.347245693206787,
2118
+ "learning_rate": 3.695134700614372e-05,
2119
+ "loss": 0.06514678001403809,
2120
+ "step": 1275,
2121
+ "token_acc": 0.9790836653386454
2122
+ },
2123
+ {
2124
+ "epoch": 2.826804835924007,
2125
+ "grad_norm": 3.311340808868408,
2126
+ "learning_rate": 3.636504538001882e-05,
2127
+ "loss": 0.0548922598361969,
2128
+ "step": 1280,
2129
+ "token_acc": 0.9831432821021319
2130
+ },
2131
+ {
2132
+ "epoch": 2.8378583765112264,
2133
+ "grad_norm": 1.910610556602478,
2134
+ "learning_rate": 3.5780762285753616e-05,
2135
+ "loss": 0.04039471745491028,
2136
+ "step": 1285,
2137
+ "token_acc": 0.9861454725383474
2138
+ },
2139
+ {
2140
+ "epoch": 2.8489119170984454,
2141
+ "grad_norm": 3.0203163623809814,
2142
+ "learning_rate": 3.519858422118206e-05,
2143
+ "loss": 0.06901986002922059,
2144
+ "step": 1290,
2145
+ "token_acc": 0.9781312127236581
2146
+ },
2147
+ {
2148
+ "epoch": 2.859965457685665,
2149
+ "grad_norm": 7.763772487640381,
2150
+ "learning_rate": 3.461859737250752e-05,
2151
+ "loss": 0.042749127745628356,
2152
+ "step": 1295,
2153
+ "token_acc": 0.9840637450199203
2154
+ },
2155
+ {
2156
+ "epoch": 2.871018998272884,
2157
+ "grad_norm": 4.355055332183838,
2158
+ "learning_rate": 3.4040887601543574e-05,
2159
+ "loss": 0.06063474416732788,
2160
+ "step": 1300,
2161
+ "token_acc": 0.9815277084373439
2162
+ },
2163
+ {
2164
+ "epoch": 2.8820725388601036,
2165
+ "grad_norm": 3.254500389099121,
2166
+ "learning_rate": 3.346554043300308e-05,
2167
+ "loss": 0.058100783824920656,
2168
+ "step": 1305,
2169
+ "token_acc": 0.9791976225854383
2170
+ },
2171
+ {
2172
+ "epoch": 2.893126079447323,
2173
+ "grad_norm": 4.576897621154785,
2174
+ "learning_rate": 3.289264104183691e-05,
2175
+ "loss": 0.05097652673721313,
2176
+ "step": 1310,
2177
+ "token_acc": 0.983201581027668
2178
+ },
2179
+ {
2180
+ "epoch": 2.9041796200345424,
2181
+ "grad_norm": 4.716442584991455,
2182
+ "learning_rate": 3.232227424062464e-05,
2183
+ "loss": 0.05045266747474671,
2184
+ "step": 1315,
2185
+ "token_acc": 0.9829059829059829
2186
+ },
2187
+ {
2188
+ "epoch": 2.915233160621762,
2189
+ "grad_norm": 3.8253066539764404,
2190
+ "learning_rate": 3.175452446701873e-05,
2191
+ "loss": 0.05482856035232544,
2192
+ "step": 1320,
2193
+ "token_acc": 0.9791356184798807
2194
+ },
2195
+ {
2196
+ "epoch": 2.926286701208981,
2197
+ "grad_norm": 4.78739595413208,
2198
+ "learning_rate": 3.118947577124439e-05,
2199
+ "loss": 0.056392842531204225,
2200
+ "step": 1325,
2201
+ "token_acc": 0.9797130133597229
2202
+ },
2203
+ {
2204
+ "epoch": 2.9373402417962002,
2205
+ "grad_norm": 2.2553937435150146,
2206
+ "learning_rate": 3.062721180365669e-05,
2207
+ "loss": 0.05316250324249268,
2208
+ "step": 1330,
2209
+ "token_acc": 0.9816831683168317
2210
+ },
2211
+ {
2212
+ "epoch": 2.9483937823834196,
2213
+ "grad_norm": 3.2007408142089844,
2214
+ "learning_rate": 3.0067815802356714e-05,
2215
+ "loss": 0.055870598554611205,
2216
+ "step": 1335,
2217
+ "token_acc": 0.9796526054590571
2218
+ },
2219
+ {
2220
+ "epoch": 2.959447322970639,
2221
+ "grad_norm": 1.5691827535629272,
2222
+ "learning_rate": 2.9511370580869213e-05,
2223
+ "loss": 0.04847137331962585,
2224
+ "step": 1340,
2225
+ "token_acc": 0.980635551142006
2226
+ },
2227
+ {
2228
+ "epoch": 2.9705008635578585,
2229
+ "grad_norm": 4.0064826011657715,
2230
+ "learning_rate": 2.895795851588252e-05,
2231
+ "loss": 0.061286211013793945,
2232
+ "step": 1345,
2233
+ "token_acc": 0.9805583250249252
2234
+ },
2235
+ {
2236
+ "epoch": 2.981554404145078,
2237
+ "grad_norm": 3.6499500274658203,
2238
+ "learning_rate": 2.8407661535053588e-05,
2239
+ "loss": 0.0678468644618988,
2240
+ "step": 1350,
2241
+ "token_acc": 0.9766401590457257
2242
+ },
2243
+ {
2244
+ "epoch": 2.9926079447322973,
2245
+ "grad_norm": 3.093632936477661,
2246
+ "learning_rate": 2.7860561104879357e-05,
2247
+ "loss": 0.04808221161365509,
2248
+ "step": 1355,
2249
+ "token_acc": 0.9815920398009951
2250
+ },
2251
+ {
2252
+ "epoch": 3.0,
2253
+ "eval_loss": 0.057897526770830154,
2254
+ "eval_runtime": 2238.3251,
2255
+ "eval_samples_per_second": 1.049,
2256
+ "eval_steps_per_second": 1.049,
2257
+ "eval_token_acc": 0.9798328426894055,
2258
+ "step": 1359
2259
+ }
2260
+ ],
2261
+ "logging_steps": 5,
2262
+ "max_steps": 1359,
2263
+ "num_input_tokens_seen": 0,
2264
+ "num_train_epochs": 3,
2265
+ "save_steps": 200,
2266
+ "stateful_callbacks": {
2267
+ "TrainerControl": {
2268
+ "args": {
2269
+ "should_epoch_stop": false,
2270
+ "should_evaluate": false,
2271
+ "should_log": false,
2272
+ "should_save": true,
2273
+ "should_training_stop": true
2274
+ },
2275
+ "attributes": {}
2276
+ }
2277
+ },
2278
+ "total_flos": 8.83259479920255e+17,
2279
+ "train_batch_size": 1,
2280
+ "trial_name": null,
2281
+ "trial_params": null
2282
+ }