Prummn commited on
Commit
844f5ab
·
verified ·
1 Parent(s): 78b1b65

Upload full run folder: v0-20260123-013837

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. v0-20260123-014416/args.json +351 -0
  2. v0-20260123-014416/checkpoint-2000/README.md +202 -0
  3. v0-20260123-014416/checkpoint-2000/adapter_config.json +26 -0
  4. v0-20260123-014416/checkpoint-2000/adapter_model.safetensors +3 -0
  5. v0-20260123-014416/checkpoint-2000/additional_config.json +1 -0
  6. v0-20260123-014416/checkpoint-2000/args.json +351 -0
  7. v0-20260123-014416/checkpoint-2000/optimizer.pt +3 -0
  8. v0-20260123-014416/checkpoint-2000/rng_state.pth +3 -0
  9. v0-20260123-014416/checkpoint-2000/scheduler.pt +3 -0
  10. v0-20260123-014416/checkpoint-2000/trainer_state.json +210 -0
  11. v0-20260123-014416/checkpoint-2000/training_args.bin +3 -0
  12. v0-20260123-014416/checkpoint-4000/README.md +202 -0
  13. v0-20260123-014416/checkpoint-4000/adapter_config.json +26 -0
  14. v0-20260123-014416/checkpoint-4000/adapter_model.safetensors +3 -0
  15. v0-20260123-014416/checkpoint-4000/additional_config.json +1 -0
  16. v0-20260123-014416/checkpoint-4000/args.json +351 -0
  17. v0-20260123-014416/checkpoint-4000/optimizer.pt +3 -0
  18. v0-20260123-014416/checkpoint-4000/rng_state.pth +3 -0
  19. v0-20260123-014416/checkpoint-4000/scheduler.pt +3 -0
  20. v0-20260123-014416/checkpoint-4000/trainer_state.json +379 -0
  21. v0-20260123-014416/checkpoint-4000/training_args.bin +3 -0
  22. v0-20260123-014416/logging.jsonl +44 -0
  23. v0-20260123-014416/v0-20260123-194351/args.json +351 -0
  24. v0-20260123-014416/v0-20260123-194351/logging.jsonl +1 -0
  25. v0-20260123-014416/v1-20260123-200453/args.json +351 -0
  26. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/README.md +202 -0
  27. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_config.json +26 -0
  28. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_model.safetensors +3 -0
  29. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/additional_config.json +1 -0
  30. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/args.json +351 -0
  31. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/optimizer.pt +3 -0
  32. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/rng_state.pth +3 -0
  33. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/scheduler.pt +3 -0
  34. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/trainer_state.json +548 -0
  35. v0-20260123-014416/v1-20260123-200453/checkpoint-6000/training_args.bin +3 -0
  36. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/README.md +202 -0
  37. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_config.json +26 -0
  38. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_model.safetensors +3 -0
  39. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/additional_config.json +1 -0
  40. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/args.json +351 -0
  41. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/optimizer.pt +3 -0
  42. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/rng_state.pth +3 -0
  43. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/scheduler.pt +3 -0
  44. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/trainer_state.json +717 -0
  45. v0-20260123-014416/v1-20260123-200453/checkpoint-8000/training_args.bin +3 -0
  46. v0-20260123-014416/v1-20260123-200453/checkpoint-9358/README.md +202 -0
  47. v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_config.json +26 -0
  48. v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_model.safetensors +3 -0
  49. v0-20260123-014416/v1-20260123-200453/checkpoint-9358/additional_config.json +1 -0
  50. v0-20260123-014416/v1-20260123-200453/checkpoint-9358/args.json +351 -0
v0-20260123-014416/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": null,
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7f32c017c4c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/checkpoint-2000/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
v0-20260123-014416/checkpoint-2000/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [],
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": "^(encoder.*\\.(query|mlp.0|mlp.2|out|value|key)|adapter.*\\.(linear1|linear2))$",
23
+ "task_type": "CAUSAL_LM",
24
+ "use_dora": false,
25
+ "use_rslora": false
26
+ }
v0-20260123-014416/checkpoint-2000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be7279fd3005dd01721976209db91a2a111273879d3a23d48b36ec7a306aed3e
3
+ size 11988912
v0-20260123-014416/checkpoint-2000/additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": null, "lorap_emb_lr": 1e-06}
v0-20260123-014416/checkpoint-2000/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": null,
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7f32c017c4c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/checkpoint-2000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2efe81479a806272b0f65c224098b8615b114494e4739155437b68403abf17f9
3
+ size 24205999
v0-20260123-014416/checkpoint-2000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba831bcb8faf9f5101c94e9dfbf8557a87bf40c6ea33f78a837bdfcb9953b0a3
3
+ size 14645
v0-20260123-014416/checkpoint-2000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24cc5c369df9f5313e40bf46692e1978487bd2ab62aa6773313c29be84334c1b
3
+ size 1465
v0-20260123-014416/checkpoint-2000/trainer_state.json ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.55640507,
3
+ "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-2000",
4
+ "epoch": 0.21372088053002777,
5
+ "eval_steps": 2000,
6
+ "global_step": 2000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.00010686044026501389,
13
+ "grad_norm": 1.6953125,
14
+ "learning_rate": 2.6595744680851065e-08,
15
+ "loss": 0.7167500853538513,
16
+ "step": 1,
17
+ "token_acc": 0.8417487684729064
18
+ },
19
+ {
20
+ "epoch": 0.010686044026501388,
21
+ "grad_norm": 1.4453125,
22
+ "learning_rate": 2.6595744680851065e-06,
23
+ "loss": 0.6973034637142913,
24
+ "step": 100,
25
+ "token_acc": 0.8467176205711577
26
+ },
27
+ {
28
+ "epoch": 0.021372088053002777,
29
+ "grad_norm": 1.046875,
30
+ "learning_rate": 4.999978873236658e-06,
31
+ "loss": 0.6577812957763672,
32
+ "step": 200,
33
+ "token_acc": 0.8531807124137688
34
+ },
35
+ {
36
+ "epoch": 0.03205813207950417,
37
+ "grad_norm": 1.328125,
38
+ "learning_rate": 4.998159847367393e-06,
39
+ "loss": 0.6022152709960937,
40
+ "step": 300,
41
+ "token_acc": 0.8620946875218537
42
+ },
43
+ {
44
+ "epoch": 0.04274417610600555,
45
+ "grad_norm": 1.2421875,
46
+ "learning_rate": 4.993408991262632e-06,
47
+ "loss": 0.5575482177734375,
48
+ "step": 400,
49
+ "token_acc": 0.8680139909957781
50
+ },
51
+ {
52
+ "epoch": 0.05343022013250694,
53
+ "grad_norm": 1.8984375,
54
+ "learning_rate": 4.985731880507775e-06,
55
+ "loss": 0.5264178466796875,
56
+ "step": 500,
57
+ "token_acc": 0.8696029378557931
58
+ },
59
+ {
60
+ "epoch": 0.06411626415900834,
61
+ "grad_norm": 1.1171875,
62
+ "learning_rate": 4.975137524928774e-06,
63
+ "loss": 0.4771532440185547,
64
+ "step": 600,
65
+ "token_acc": 0.8748165713816137
66
+ },
67
+ {
68
+ "epoch": 0.07480230818550973,
69
+ "grad_norm": 1.4765625,
70
+ "learning_rate": 4.961638358018247e-06,
71
+ "loss": 0.4643104553222656,
72
+ "step": 700,
73
+ "token_acc": 0.8754063671076119
74
+ },
75
+ {
76
+ "epoch": 0.0854883522120111,
77
+ "grad_norm": 1.109375,
78
+ "learning_rate": 4.945250222343571e-06,
79
+ "loss": 0.4561585235595703,
80
+ "step": 800,
81
+ "token_acc": 0.8795579017210171
82
+ },
83
+ {
84
+ "epoch": 0.0961743962385125,
85
+ "grad_norm": 1.171875,
86
+ "learning_rate": 4.925992350954112e-06,
87
+ "loss": 0.4534299850463867,
88
+ "step": 900,
89
+ "token_acc": 0.8851879828938838
90
+ },
91
+ {
92
+ "epoch": 0.10686044026501389,
93
+ "grad_norm": 1.171875,
94
+ "learning_rate": 4.903887344809403e-06,
95
+ "loss": 0.42084632873535155,
96
+ "step": 1000,
97
+ "token_acc": 0.8909899302321495
98
+ },
99
+ {
100
+ "epoch": 0.11754648429151528,
101
+ "grad_norm": 1.390625,
102
+ "learning_rate": 4.8789611462547414e-06,
103
+ "loss": 0.43503849029541014,
104
+ "step": 1100,
105
+ "token_acc": 0.8900309663514769
106
+ },
107
+ {
108
+ "epoch": 0.12823252831801668,
109
+ "grad_norm": 1.21875,
110
+ "learning_rate": 4.851243008575364e-06,
111
+ "loss": 0.4078071594238281,
112
+ "step": 1200,
113
+ "token_acc": 0.8972086577408744
114
+ },
115
+ {
116
+ "epoch": 0.13891857234451807,
117
+ "grad_norm": 1.3046875,
118
+ "learning_rate": 4.8207654616649035e-06,
119
+ "loss": 0.4169163131713867,
120
+ "step": 1300,
121
+ "token_acc": 0.8944712980277317
122
+ },
123
+ {
124
+ "epoch": 0.14960461637101946,
125
+ "grad_norm": 1.25,
126
+ "learning_rate": 4.7875642738484475e-06,
127
+ "loss": 0.41801651000976564,
128
+ "step": 1400,
129
+ "token_acc": 0.8954553844138678
130
+ },
131
+ {
132
+ "epoch": 0.16029066039752085,
133
+ "grad_norm": 1.0390625,
134
+ "learning_rate": 4.7516784099049705e-06,
135
+ "loss": 0.42521297454833984,
136
+ "step": 1500,
137
+ "token_acc": 0.8927475600806661
138
+ },
139
+ {
140
+ "epoch": 0.1709767044240222,
141
+ "grad_norm": 1.0625,
142
+ "learning_rate": 4.713149985338428e-06,
143
+ "loss": 0.40242053985595705,
144
+ "step": 1600,
145
+ "token_acc": 0.8985869393070196
146
+ },
147
+ {
148
+ "epoch": 0.1816627484505236,
149
+ "grad_norm": 1.6953125,
150
+ "learning_rate": 4.672024216951176e-06,
151
+ "loss": 0.39753116607666017,
152
+ "step": 1700,
153
+ "token_acc": 0.9003083728795708
154
+ },
155
+ {
156
+ "epoch": 0.192348792477025,
157
+ "grad_norm": 1.328125,
158
+ "learning_rate": 4.628349369777712e-06,
159
+ "loss": 0.40670780181884764,
160
+ "step": 1800,
161
+ "token_acc": 0.8985113659223496
162
+ },
163
+ {
164
+ "epoch": 0.20303483650352638,
165
+ "grad_norm": 1.609375,
166
+ "learning_rate": 4.58217670044103e-06,
167
+ "loss": 0.4040363311767578,
168
+ "step": 1900,
169
+ "token_acc": 0.898509424415663
170
+ },
171
+ {
172
+ "epoch": 0.21372088053002777,
173
+ "grad_norm": 1.3203125,
174
+ "learning_rate": 4.533560396998055e-06,
175
+ "loss": 0.3897912216186523,
176
+ "step": 2000,
177
+ "token_acc": 0.9022499382853919
178
+ },
179
+ {
180
+ "epoch": 0.21372088053002777,
181
+ "eval_loss": 0.5564050674438477,
182
+ "eval_runtime": 1628.334,
183
+ "eval_samples_per_second": 20.433,
184
+ "eval_steps_per_second": 1.277,
185
+ "eval_token_acc": 0.8750522224756502,
186
+ "step": 2000
187
+ }
188
+ ],
189
+ "logging_steps": 100,
190
+ "max_steps": 9358,
191
+ "num_input_tokens_seen": 0,
192
+ "num_train_epochs": 1,
193
+ "save_steps": 2000,
194
+ "stateful_callbacks": {
195
+ "TrainerControl": {
196
+ "args": {
197
+ "should_epoch_stop": false,
198
+ "should_evaluate": false,
199
+ "should_log": false,
200
+ "should_save": true,
201
+ "should_training_stop": false
202
+ },
203
+ "attributes": {}
204
+ }
205
+ },
206
+ "total_flos": 1.607272128827302e+18,
207
+ "train_batch_size": 16,
208
+ "trial_name": null,
209
+ "trial_params": null
210
+ }
v0-20260123-014416/checkpoint-2000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e61a5ccecfe4850e11b9fb1b8bea0737456f069a42a114f94ef68317432e4e8
3
+ size 7185
v0-20260123-014416/checkpoint-4000/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
v0-20260123-014416/checkpoint-4000/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [],
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": "^(encoder.*\\.(query|mlp.0|mlp.2|out|value|key)|adapter.*\\.(linear1|linear2))$",
23
+ "task_type": "CAUSAL_LM",
24
+ "use_dora": false,
25
+ "use_rslora": false
26
+ }
v0-20260123-014416/checkpoint-4000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f315dcc36f8a2d56b54570ce77e301a8a30c57d4fd0824ce5922f44bab8ee778
3
+ size 11988912
v0-20260123-014416/checkpoint-4000/additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": null, "lorap_emb_lr": 1e-06}
v0-20260123-014416/checkpoint-4000/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": null,
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7f32c017c4c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/checkpoint-4000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b3ddf443d798031411e6f911b39d2e0b74db8a9e149794dafbaa01adc706b09
3
+ size 24205999
v0-20260123-014416/checkpoint-4000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65a0256738d0a2b4fb93409e385a10b55cf44ea01734625f77b31f99a543f3a1
3
+ size 14645
v0-20260123-014416/checkpoint-4000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5463e23ca1127ec48142e74ff51708175807ecfcea702865fb9c8596fa6ce8c0
3
+ size 1465
v0-20260123-014416/checkpoint-4000/trainer_state.json ADDED
@@ -0,0 +1,379 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.55598271,
3
+ "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
4
+ "epoch": 0.42744176106005555,
5
+ "eval_steps": 2000,
6
+ "global_step": 4000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.00010686044026501389,
13
+ "grad_norm": 1.6953125,
14
+ "learning_rate": 2.6595744680851065e-08,
15
+ "loss": 0.7167500853538513,
16
+ "step": 1,
17
+ "token_acc": 0.8417487684729064
18
+ },
19
+ {
20
+ "epoch": 0.010686044026501388,
21
+ "grad_norm": 1.4453125,
22
+ "learning_rate": 2.6595744680851065e-06,
23
+ "loss": 0.6973034637142913,
24
+ "step": 100,
25
+ "token_acc": 0.8467176205711577
26
+ },
27
+ {
28
+ "epoch": 0.021372088053002777,
29
+ "grad_norm": 1.046875,
30
+ "learning_rate": 4.999978873236658e-06,
31
+ "loss": 0.6577812957763672,
32
+ "step": 200,
33
+ "token_acc": 0.8531807124137688
34
+ },
35
+ {
36
+ "epoch": 0.03205813207950417,
37
+ "grad_norm": 1.328125,
38
+ "learning_rate": 4.998159847367393e-06,
39
+ "loss": 0.6022152709960937,
40
+ "step": 300,
41
+ "token_acc": 0.8620946875218537
42
+ },
43
+ {
44
+ "epoch": 0.04274417610600555,
45
+ "grad_norm": 1.2421875,
46
+ "learning_rate": 4.993408991262632e-06,
47
+ "loss": 0.5575482177734375,
48
+ "step": 400,
49
+ "token_acc": 0.8680139909957781
50
+ },
51
+ {
52
+ "epoch": 0.05343022013250694,
53
+ "grad_norm": 1.8984375,
54
+ "learning_rate": 4.985731880507775e-06,
55
+ "loss": 0.5264178466796875,
56
+ "step": 500,
57
+ "token_acc": 0.8696029378557931
58
+ },
59
+ {
60
+ "epoch": 0.06411626415900834,
61
+ "grad_norm": 1.1171875,
62
+ "learning_rate": 4.975137524928774e-06,
63
+ "loss": 0.4771532440185547,
64
+ "step": 600,
65
+ "token_acc": 0.8748165713816137
66
+ },
67
+ {
68
+ "epoch": 0.07480230818550973,
69
+ "grad_norm": 1.4765625,
70
+ "learning_rate": 4.961638358018247e-06,
71
+ "loss": 0.4643104553222656,
72
+ "step": 700,
73
+ "token_acc": 0.8754063671076119
74
+ },
75
+ {
76
+ "epoch": 0.0854883522120111,
77
+ "grad_norm": 1.109375,
78
+ "learning_rate": 4.945250222343571e-06,
79
+ "loss": 0.4561585235595703,
80
+ "step": 800,
81
+ "token_acc": 0.8795579017210171
82
+ },
83
+ {
84
+ "epoch": 0.0961743962385125,
85
+ "grad_norm": 1.171875,
86
+ "learning_rate": 4.925992350954112e-06,
87
+ "loss": 0.4534299850463867,
88
+ "step": 900,
89
+ "token_acc": 0.8851879828938838
90
+ },
91
+ {
92
+ "epoch": 0.10686044026501389,
93
+ "grad_norm": 1.171875,
94
+ "learning_rate": 4.903887344809403e-06,
95
+ "loss": 0.42084632873535155,
96
+ "step": 1000,
97
+ "token_acc": 0.8909899302321495
98
+ },
99
+ {
100
+ "epoch": 0.11754648429151528,
101
+ "grad_norm": 1.390625,
102
+ "learning_rate": 4.8789611462547414e-06,
103
+ "loss": 0.43503849029541014,
104
+ "step": 1100,
105
+ "token_acc": 0.8900309663514769
106
+ },
107
+ {
108
+ "epoch": 0.12823252831801668,
109
+ "grad_norm": 1.21875,
110
+ "learning_rate": 4.851243008575364e-06,
111
+ "loss": 0.4078071594238281,
112
+ "step": 1200,
113
+ "token_acc": 0.8972086577408744
114
+ },
115
+ {
116
+ "epoch": 0.13891857234451807,
117
+ "grad_norm": 1.3046875,
118
+ "learning_rate": 4.8207654616649035e-06,
119
+ "loss": 0.4169163131713867,
120
+ "step": 1300,
121
+ "token_acc": 0.8944712980277317
122
+ },
123
+ {
124
+ "epoch": 0.14960461637101946,
125
+ "grad_norm": 1.25,
126
+ "learning_rate": 4.7875642738484475e-06,
127
+ "loss": 0.41801651000976564,
128
+ "step": 1400,
129
+ "token_acc": 0.8954553844138678
130
+ },
131
+ {
132
+ "epoch": 0.16029066039752085,
133
+ "grad_norm": 1.0390625,
134
+ "learning_rate": 4.7516784099049705e-06,
135
+ "loss": 0.42521297454833984,
136
+ "step": 1500,
137
+ "token_acc": 0.8927475600806661
138
+ },
139
+ {
140
+ "epoch": 0.1709767044240222,
141
+ "grad_norm": 1.0625,
142
+ "learning_rate": 4.713149985338428e-06,
143
+ "loss": 0.40242053985595705,
144
+ "step": 1600,
145
+ "token_acc": 0.8985869393070196
146
+ },
147
+ {
148
+ "epoch": 0.1816627484505236,
149
+ "grad_norm": 1.6953125,
150
+ "learning_rate": 4.672024216951176e-06,
151
+ "loss": 0.39753116607666017,
152
+ "step": 1700,
153
+ "token_acc": 0.9003083728795708
154
+ },
155
+ {
156
+ "epoch": 0.192348792477025,
157
+ "grad_norm": 1.328125,
158
+ "learning_rate": 4.628349369777712e-06,
159
+ "loss": 0.40670780181884764,
160
+ "step": 1800,
161
+ "token_acc": 0.8985113659223496
162
+ },
163
+ {
164
+ "epoch": 0.20303483650352638,
165
+ "grad_norm": 1.609375,
166
+ "learning_rate": 4.58217670044103e-06,
167
+ "loss": 0.4040363311767578,
168
+ "step": 1900,
169
+ "token_acc": 0.898509424415663
170
+ },
171
+ {
172
+ "epoch": 0.21372088053002777,
173
+ "grad_norm": 1.3203125,
174
+ "learning_rate": 4.533560396998055e-06,
175
+ "loss": 0.3897912216186523,
176
+ "step": 2000,
177
+ "token_acc": 0.9022499382853919
178
+ },
179
+ {
180
+ "epoch": 0.21372088053002777,
181
+ "eval_loss": 0.5564050674438477,
182
+ "eval_runtime": 1628.334,
183
+ "eval_samples_per_second": 20.433,
184
+ "eval_steps_per_second": 1.277,
185
+ "eval_token_acc": 0.8750522224756502,
186
+ "step": 2000
187
+ },
188
+ {
189
+ "epoch": 0.22440692455652916,
190
+ "grad_norm": 1.390625,
191
+ "learning_rate": 4.482557515344757e-06,
192
+ "loss": 0.39870925903320314,
193
+ "step": 2100,
194
+ "token_acc": 0.8990360044410765
195
+ },
196
+ {
197
+ "epoch": 0.23509296858303055,
198
+ "grad_norm": 1.140625,
199
+ "learning_rate": 4.429227912255589e-06,
200
+ "loss": 0.38395835876464846,
201
+ "step": 2200,
202
+ "token_acc": 0.902773223403757
203
+ },
204
+ {
205
+ "epoch": 0.24577901260953194,
206
+ "grad_norm": 0.96484375,
207
+ "learning_rate": 4.3736341751358166e-06,
208
+ "loss": 0.39471343994140623,
209
+ "step": 2300,
210
+ "token_acc": 0.9010956332106893
211
+ },
212
+ {
213
+ "epoch": 0.25646505663603336,
214
+ "grad_norm": 1.4140625,
215
+ "learning_rate": 4.315841548569193e-06,
216
+ "loss": 0.39796932220458986,
217
+ "step": 2400,
218
+ "token_acc": 0.9007656570069128
219
+ },
220
+ {
221
+ "epoch": 0.2671511006625347,
222
+ "grad_norm": 1.265625,
223
+ "learning_rate": 4.255917857747183e-06,
224
+ "loss": 0.3913582992553711,
225
+ "step": 2500,
226
+ "token_acc": 0.9015067434468572
227
+ },
228
+ {
229
+ "epoch": 0.27783714468903614,
230
+ "grad_norm": 1.46875,
231
+ "learning_rate": 4.193933428869592e-06,
232
+ "loss": 0.38879886627197263,
233
+ "step": 2600,
234
+ "token_acc": 0.9019667027692453
235
+ },
236
+ {
237
+ "epoch": 0.2885231887155375,
238
+ "grad_norm": 2.4375,
239
+ "learning_rate": 4.129961006610028e-06,
240
+ "loss": 0.3972347259521484,
241
+ "step": 2700,
242
+ "token_acc": 0.9001422292689183
243
+ },
244
+ {
245
+ "epoch": 0.2992092327420389,
246
+ "grad_norm": 1.2578125,
247
+ "learning_rate": 4.064075668743043e-06,
248
+ "loss": 0.3885000991821289,
249
+ "step": 2800,
250
+ "token_acc": 0.9027788505528588
251
+ },
252
+ {
253
+ "epoch": 0.3098952767685403,
254
+ "grad_norm": 1.265625,
255
+ "learning_rate": 3.996354738033166e-06,
256
+ "loss": 0.39520988464355467,
257
+ "step": 2900,
258
+ "token_acc": 0.9011279097672186
259
+ },
260
+ {
261
+ "epoch": 0.3205813207950417,
262
+ "grad_norm": 0.921875,
263
+ "learning_rate": 3.926877691489222e-06,
264
+ "loss": 0.3871217346191406,
265
+ "step": 3000,
266
+ "token_acc": 0.9022083313787118
267
+ },
268
+ {
269
+ "epoch": 0.33126736482154306,
270
+ "grad_norm": 1.375,
271
+ "learning_rate": 3.855726067090433e-06,
272
+ "loss": 0.3810308074951172,
273
+ "step": 3100,
274
+ "token_acc": 0.9034641205430405
275
+ },
276
+ {
277
+ "epoch": 0.3419534088480444,
278
+ "grad_norm": 1.5546875,
279
+ "learning_rate": 3.7829833680937804e-06,
280
+ "loss": 0.37891117095947263,
281
+ "step": 3200,
282
+ "token_acc": 0.9025715320536037
283
+ },
284
+ {
285
+ "epoch": 0.35263945287454584,
286
+ "grad_norm": 1.59375,
287
+ "learning_rate": 3.7087349650349234e-06,
288
+ "loss": 0.40724864959716794,
289
+ "step": 3300,
290
+ "token_acc": 0.8990014549892049
291
+ },
292
+ {
293
+ "epoch": 0.3633254969010472,
294
+ "grad_norm": 1.1171875,
295
+ "learning_rate": 3.6330679955376843e-06,
296
+ "loss": 0.39273796081542967,
297
+ "step": 3400,
298
+ "token_acc": 0.900933421493117
299
+ },
300
+ {
301
+ "epoch": 0.3740115409275486,
302
+ "grad_norm": 1.5,
303
+ "learning_rate": 3.556071262049684e-06,
304
+ "loss": 0.3975730514526367,
305
+ "step": 3500,
306
+ "token_acc": 0.8998019157822584
307
+ },
308
+ {
309
+ "epoch": 0.38469758495405,
310
+ "grad_norm": 2.140625,
311
+ "learning_rate": 3.477835127624158e-06,
312
+ "loss": 0.37645999908447264,
313
+ "step": 3600,
314
+ "token_acc": 0.9051004274755519
315
+ },
316
+ {
317
+ "epoch": 0.3953836289805514,
318
+ "grad_norm": 1.1796875,
319
+ "learning_rate": 3.39845140987024e-06,
320
+ "loss": 0.4185586166381836,
321
+ "step": 3700,
322
+ "token_acc": 0.8963374986700084
323
+ },
324
+ {
325
+ "epoch": 0.40606967300705277,
326
+ "grad_norm": 1.0390625,
327
+ "learning_rate": 3.318013273196187e-06,
328
+ "loss": 0.3819176483154297,
329
+ "step": 3800,
330
+ "token_acc": 0.9037970813391503
331
+ },
332
+ {
333
+ "epoch": 0.4167557170335542,
334
+ "grad_norm": 0.8828125,
335
+ "learning_rate": 3.2366151194720163e-06,
336
+ "loss": 0.4014278793334961,
337
+ "step": 3900,
338
+ "token_acc": 0.8989897804049096
339
+ },
340
+ {
341
+ "epoch": 0.42744176106005555,
342
+ "grad_norm": 1.2265625,
343
+ "learning_rate": 3.1543524772398464e-06,
344
+ "loss": 0.3903131103515625,
345
+ "step": 4000,
346
+ "token_acc": 0.9023779507523624
347
+ },
348
+ {
349
+ "epoch": 0.42744176106005555,
350
+ "eval_loss": 0.5559827089309692,
351
+ "eval_runtime": 1625.5851,
352
+ "eval_samples_per_second": 20.468,
353
+ "eval_steps_per_second": 1.28,
354
+ "eval_token_acc": 0.8759747254509199,
355
+ "step": 4000
356
+ }
357
+ ],
358
+ "logging_steps": 100,
359
+ "max_steps": 9358,
360
+ "num_input_tokens_seen": 0,
361
+ "num_train_epochs": 1,
362
+ "save_steps": 2000,
363
+ "stateful_callbacks": {
364
+ "TrainerControl": {
365
+ "args": {
366
+ "should_epoch_stop": false,
367
+ "should_evaluate": false,
368
+ "should_log": false,
369
+ "should_save": true,
370
+ "should_training_stop": false
371
+ },
372
+ "attributes": {}
373
+ }
374
+ },
375
+ "total_flos": 3.218244349496107e+18,
376
+ "train_batch_size": 16,
377
+ "trial_name": null,
378
+ "trial_params": null
379
+ }
v0-20260123-014416/checkpoint-4000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e61a5ccecfe4850e11b9fb1b8bea0737456f069a42a114f94ef68317432e4e8
3
+ size 7185
v0-20260123-014416/logging.jsonl ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"loss": 0.71675009, "grad_norm": 1.6953125, "learning_rate": 3e-08, "token_acc": 0.84174877, "epoch": 0.00010686, "global_step/max_steps": "1/9358", "percentage": "0.01%", "elapsed_time": "11s", "remaining_time": "1d 6h 38m 27s", "memory(GiB)": 35.58, "train_speed(iter/s)": 0.084826}
2
+ {"loss": 0.69730346, "grad_norm": 1.4453125, "learning_rate": 2.66e-06, "token_acc": 0.84671762, "epoch": 0.01068604, "global_step/max_steps": "100/9358", "percentage": "1.07%", "elapsed_time": "16m 25s", "remaining_time": "1d 1h 20m 48s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101459}
3
+ {"loss": 0.6577813, "grad_norm": 1.046875, "learning_rate": 5e-06, "token_acc": 0.85318071, "epoch": 0.02137209, "global_step/max_steps": "200/9358", "percentage": "2.14%", "elapsed_time": "32m 50s", "remaining_time": "1d 1h 3m 50s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101495}
4
+ {"loss": 0.60221527, "grad_norm": 1.328125, "learning_rate": 5e-06, "token_acc": 0.86209469, "epoch": 0.03205813, "global_step/max_steps": "300/9358", "percentage": "3.21%", "elapsed_time": "49m 14s", "remaining_time": "1d 0h 46m 52s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101533}
5
+ {"loss": 0.55754822, "grad_norm": 1.2421875, "learning_rate": 4.99e-06, "token_acc": 0.86801399, "epoch": 0.04274418, "global_step/max_steps": "400/9358", "percentage": "4.27%", "elapsed_time": "1h 5m 36s", "remaining_time": "1d 0h 29m 27s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101603}
6
+ {"loss": 0.52641785, "grad_norm": 1.8984375, "learning_rate": 4.99e-06, "token_acc": 0.86960294, "epoch": 0.05343022, "global_step/max_steps": "500/9358", "percentage": "5.34%", "elapsed_time": "1h 21m 59s", "remaining_time": "1d 0h 12m 32s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101638}
7
+ {"loss": 0.47715324, "grad_norm": 1.1171875, "learning_rate": 4.98e-06, "token_acc": 0.87481657, "epoch": 0.06411626, "global_step/max_steps": "600/9358", "percentage": "6.41%", "elapsed_time": "1h 38m 27s", "remaining_time": "23h 57m 16s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101558}
8
+ {"loss": 0.46431046, "grad_norm": 1.4765625, "learning_rate": 4.96e-06, "token_acc": 0.87540637, "epoch": 0.07480231, "global_step/max_steps": "700/9358", "percentage": "7.48%", "elapsed_time": "1h 54m 47s", "remaining_time": "23h 39m 42s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.10164}
9
+ {"loss": 0.45615852, "grad_norm": 1.109375, "learning_rate": 4.95e-06, "token_acc": 0.8795579, "epoch": 0.08548835, "global_step/max_steps": "800/9358", "percentage": "8.55%", "elapsed_time": "2h 11m 18s", "remaining_time": "23h 24m 39s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101543}
10
+ {"loss": 0.45342999, "grad_norm": 1.171875, "learning_rate": 4.93e-06, "token_acc": 0.88518798, "epoch": 0.0961744, "global_step/max_steps": "900/9358", "percentage": "9.62%", "elapsed_time": "2h 27m 38s", "remaining_time": "23h 7m 27s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101601}
11
+ {"loss": 0.42084633, "grad_norm": 1.171875, "learning_rate": 4.9e-06, "token_acc": 0.89098993, "epoch": 0.10686044, "global_step/max_steps": "1000/9358", "percentage": "10.69%", "elapsed_time": "2h 44m 3s", "remaining_time": "22h 51m 9s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101593}
12
+ {"loss": 0.43503849, "grad_norm": 1.390625, "learning_rate": 4.88e-06, "token_acc": 0.89003097, "epoch": 0.11754648, "global_step/max_steps": "1100/9358", "percentage": "11.75%", "elapsed_time": "3h 0m 28s", "remaining_time": "22h 34m 52s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101583}
13
+ {"loss": 0.40780716, "grad_norm": 1.21875, "learning_rate": 4.85e-06, "token_acc": 0.89720866, "epoch": 0.12823253, "global_step/max_steps": "1200/9358", "percentage": "12.82%", "elapsed_time": "3h 16m 40s", "remaining_time": "22h 17m 4s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.10169}
14
+ {"loss": 0.41691631, "grad_norm": 1.3046875, "learning_rate": 4.82e-06, "token_acc": 0.8944713, "epoch": 0.13891857, "global_step/max_steps": "1300/9358", "percentage": "13.89%", "elapsed_time": "3h 33m 2s", "remaining_time": "22h 0m 29s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101705}
15
+ {"loss": 0.41801651, "grad_norm": 1.25, "learning_rate": 4.79e-06, "token_acc": 0.89545538, "epoch": 0.14960462, "global_step/max_steps": "1400/9358", "percentage": "14.96%", "elapsed_time": "3h 49m 21s", "remaining_time": "21h 43m 44s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101733}
16
+ {"loss": 0.42521297, "grad_norm": 1.0390625, "learning_rate": 4.75e-06, "token_acc": 0.89274756, "epoch": 0.16029066, "global_step/max_steps": "1500/9358", "percentage": "16.03%", "elapsed_time": "4h 5m 54s", "remaining_time": "21h 28m 11s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101667}
17
+ {"loss": 0.40242054, "grad_norm": 1.0625, "learning_rate": 4.71e-06, "token_acc": 0.89858694, "epoch": 0.1709767, "global_step/max_steps": "1600/9358", "percentage": "17.10%", "elapsed_time": "4h 22m 22s", "remaining_time": "21h 12m 12s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101635}
18
+ {"loss": 0.39753117, "grad_norm": 1.6953125, "learning_rate": 4.67e-06, "token_acc": 0.90030837, "epoch": 0.18166275, "global_step/max_steps": "1700/9358", "percentage": "18.17%", "elapsed_time": "4h 38m 56s", "remaining_time": "20h 56m 35s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101572}
19
+ {"loss": 0.4067078, "grad_norm": 1.328125, "learning_rate": 4.63e-06, "token_acc": 0.89851137, "epoch": 0.19234879, "global_step/max_steps": "1800/9358", "percentage": "19.23%", "elapsed_time": "4h 55m 21s", "remaining_time": "20h 40m 9s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101573}
20
+ {"loss": 0.40403633, "grad_norm": 1.609375, "learning_rate": 4.58e-06, "token_acc": 0.89850942, "epoch": 0.20303484, "global_step/max_steps": "1900/9358", "percentage": "20.30%", "elapsed_time": "5h 11m 46s", "remaining_time": "20h 23m 47s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101569}
21
+ {"loss": 0.38979122, "grad_norm": 1.3203125, "learning_rate": 4.53e-06, "token_acc": 0.90224994, "epoch": 0.21372088, "global_step/max_steps": "2000/9358", "percentage": "21.37%", "elapsed_time": "5h 28m 10s", "remaining_time": "20h 7m 20s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.101573}
22
+ {"eval_loss": 0.55640507, "eval_runtime": 1628.334, "eval_samples_per_second": 20.433, "eval_steps_per_second": 1.277, "eval_token_acc": 0.87505222, "epoch": 0.21372088, "global_step/max_steps": "2000/9358", "percentage": "21.37%", "elapsed_time": "5h 55m 18s", "remaining_time": "21h 47m 11s", "memory(GiB)": 38.38, "train_speed(iter/s)": 0.093814}
23
+ {"loss": 0.39870926, "grad_norm": 1.390625, "learning_rate": 4.48e-06, "token_acc": 0.899036, "epoch": 0.22440692, "global_step/max_steps": "2100/9358", "percentage": "22.44%", "elapsed_time": "6h 11m 49s", "remaining_time": "21h 25m 4s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.094132}
24
+ {"loss": 0.38395836, "grad_norm": 1.140625, "learning_rate": 4.43e-06, "token_acc": 0.90277322, "epoch": 0.23509297, "global_step/max_steps": "2200/9358", "percentage": "23.51%", "elapsed_time": "6h 28m 24s", "remaining_time": "21h 3m 44s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.094402}
25
+ {"loss": 0.39471344, "grad_norm": 0.96484375, "learning_rate": 4.37e-06, "token_acc": 0.90109563, "epoch": 0.24577901, "global_step/max_steps": "2300/9358", "percentage": "24.58%", "elapsed_time": "6h 45m 1s", "remaining_time": "20h 42m 52s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.094646}
26
+ {"loss": 0.39796932, "grad_norm": 1.4140625, "learning_rate": 4.32e-06, "token_acc": 0.90076566, "epoch": 0.25646506, "global_step/max_steps": "2400/9358", "percentage": "25.65%", "elapsed_time": "7h 1m 23s", "remaining_time": "20h 21m 40s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.094924}
27
+ {"loss": 0.3913583, "grad_norm": 1.265625, "learning_rate": 4.26e-06, "token_acc": 0.90150674, "epoch": 0.2671511, "global_step/max_steps": "2500/9358", "percentage": "26.72%", "elapsed_time": "7h 25m 44s", "remaining_time": "20h 22m 45s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.093477}
28
+ {"loss": 0.38879887, "grad_norm": 1.46875, "learning_rate": 4.19e-06, "token_acc": 0.9019667, "epoch": 0.27783714, "global_step/max_steps": "2600/9358", "percentage": "27.78%", "elapsed_time": "7h 47m 49s", "remaining_time": "20h 15m 58s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.092628}
29
+ {"loss": 0.39723473, "grad_norm": 2.4375, "learning_rate": 4.13e-06, "token_acc": 0.90014223, "epoch": 0.28852319, "global_step/max_steps": "2700/9358", "percentage": "28.85%", "elapsed_time": "8h 5m 56s", "remaining_time": "19h 58m 17s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.092604}
30
+ {"loss": 0.3885001, "grad_norm": 1.2578125, "learning_rate": 4.06e-06, "token_acc": 0.90277885, "epoch": 0.29920923, "global_step/max_steps": "2800/9358", "percentage": "29.92%", "elapsed_time": "8h 28m 13s", "remaining_time": "19h 50m 19s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.091824}
31
+ {"loss": 0.39520988, "grad_norm": 1.265625, "learning_rate": 4e-06, "token_acc": 0.90112791, "epoch": 0.30989528, "global_step/max_steps": "2900/9358", "percentage": "30.99%", "elapsed_time": "8h 53m 15s", "remaining_time": "19h 47m 30s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.090638}
32
+ {"loss": 0.38712173, "grad_norm": 0.921875, "learning_rate": 3.93e-06, "token_acc": 0.90220833, "epoch": 0.32058132, "global_step/max_steps": "3000/9358", "percentage": "32.06%", "elapsed_time": "9h 19m 1s", "remaining_time": "19h 44m 44s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.089442}
33
+ {"loss": 0.38103081, "grad_norm": 1.375, "learning_rate": 3.86e-06, "token_acc": 0.90346412, "epoch": 0.33126736, "global_step/max_steps": "3100/9358", "percentage": "33.13%", "elapsed_time": "9h 38m 26s", "remaining_time": "19h 27m 42s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.08932}
34
+ {"loss": 0.37891117, "grad_norm": 1.5546875, "learning_rate": 3.78e-06, "token_acc": 0.90257153, "epoch": 0.34195341, "global_step/max_steps": "3200/9358", "percentage": "34.20%", "elapsed_time": "9h 54m 42s", "remaining_time": "19h 4m 27s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.089679}
35
+ {"loss": 0.40724865, "grad_norm": 1.59375, "learning_rate": 3.71e-06, "token_acc": 0.89900145, "epoch": 0.35263945, "global_step/max_steps": "3300/9358", "percentage": "35.26%", "elapsed_time": "10h 15m 21s", "remaining_time": "18h 49m 38s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.089379}
36
+ {"loss": 0.39273796, "grad_norm": 1.1171875, "learning_rate": 3.63e-06, "token_acc": 0.90093342, "epoch": 0.3633255, "global_step/max_steps": "3400/9358", "percentage": "36.33%", "elapsed_time": "10h 31m 45s", "remaining_time": "18h 27m 3s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.089697}
37
+ {"loss": 0.39757305, "grad_norm": 1.5, "learning_rate": 3.56e-06, "token_acc": 0.89980192, "epoch": 0.37401154, "global_step/max_steps": "3500/9358", "percentage": "37.40%", "elapsed_time": "10h 48m 12s", "remaining_time": "18h 4m 54s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.089992}
38
+ {"loss": 0.37646, "grad_norm": 2.140625, "learning_rate": 3.48e-06, "token_acc": 0.90510043, "epoch": 0.38469758, "global_step/max_steps": "3600/9358", "percentage": "38.47%", "elapsed_time": "11h 4m 32s", "remaining_time": "17h 42m 54s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.090287}
39
+ {"loss": 0.41855862, "grad_norm": 1.1796875, "learning_rate": 3.4e-06, "token_acc": 0.8963375, "epoch": 0.39538363, "global_step/max_steps": "3700/9358", "percentage": "39.54%", "elapsed_time": "11h 33m 43s", "remaining_time": "17h 40m 50s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.088892}
40
+ {"loss": 0.38191765, "grad_norm": 1.0390625, "learning_rate": 3.32e-06, "token_acc": 0.90379708, "epoch": 0.40606967, "global_step/max_steps": "3800/9358", "percentage": "40.61%", "elapsed_time": "12h 6m 24s", "remaining_time": "17h 42m 28s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.087187}
41
+ {"loss": 0.40142788, "grad_norm": 0.8828125, "learning_rate": 3.24e-06, "token_acc": 0.89898978, "epoch": 0.41675572, "global_step/max_steps": "3900/9358", "percentage": "41.68%", "elapsed_time": "12h 44m 43s", "remaining_time": "17h 50m 13s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.084998}
42
+ {"loss": 0.39031311, "grad_norm": 1.2265625, "learning_rate": 3.15e-06, "token_acc": 0.90237795, "epoch": 0.42744176, "global_step/max_steps": "4000/9358", "percentage": "42.74%", "elapsed_time": "13h 6m 19s", "remaining_time": "17h 33m 17s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.084782}
43
+ {"eval_loss": 0.55598271, "eval_runtime": 1625.5851, "eval_samples_per_second": 20.468, "eval_steps_per_second": 1.28, "eval_token_acc": 0.87597473, "epoch": 0.42744176, "global_step/max_steps": "4000/9358", "percentage": "42.74%", "elapsed_time": "13h 33m 25s", "remaining_time": "18h 9m 35s", "memory(GiB)": 38.6, "train_speed(iter/s)": 0.081958}
44
+ {"model_parameter_info": "PeftModelForCausalLM: 8321.1492M Params (5.9699M Trainable [0.0717%]), 0.0001M Buffers.", "last_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000", "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000", "best_metric": 0.55598271, "global_step": 4029, "log_history": [{"loss": 0.7167500853538513, "grad_norm": 1.6953125, "learning_rate": 2.6595744680851065e-08, "token_acc": 0.8417487684729064, "epoch": 0.00010686044026501389, "step": 1}, {"loss": 0.6973034637142913, "grad_norm": 1.4453125, "learning_rate": 2.6595744680851065e-06, "token_acc": 0.8467176205711577, "epoch": 0.010686044026501388, "step": 100}, {"loss": 0.6577812957763672, "grad_norm": 1.046875, "learning_rate": 4.999978873236658e-06, "token_acc": 0.8531807124137688, "epoch": 0.021372088053002777, "step": 200}, {"loss": 0.6022152709960937, "grad_norm": 1.328125, "learning_rate": 4.998159847367393e-06, "token_acc": 0.8620946875218537, "epoch": 0.03205813207950417, "step": 300}, {"loss": 0.5575482177734375, "grad_norm": 1.2421875, "learning_rate": 4.993408991262632e-06, "token_acc": 0.8680139909957781, "epoch": 0.04274417610600555, "step": 400}, {"loss": 0.5264178466796875, "grad_norm": 1.8984375, "learning_rate": 4.985731880507775e-06, "token_acc": 0.8696029378557931, "epoch": 0.05343022013250694, "step": 500}, {"loss": 0.4771532440185547, "grad_norm": 1.1171875, "learning_rate": 4.975137524928774e-06, "token_acc": 0.8748165713816137, "epoch": 0.06411626415900834, "step": 600}, {"loss": 0.4643104553222656, "grad_norm": 1.4765625, "learning_rate": 4.961638358018247e-06, "token_acc": 0.8754063671076119, "epoch": 0.07480230818550973, "step": 700}, {"loss": 0.4561585235595703, "grad_norm": 1.109375, "learning_rate": 4.945250222343571e-06, "token_acc": 0.8795579017210171, "epoch": 0.0854883522120111, "step": 800}, {"loss": 0.4534299850463867, "grad_norm": 1.171875, "learning_rate": 4.925992350954112e-06, "token_acc": 0.8851879828938838, "epoch": 0.0961743962385125, "step": 900}, {"loss": 0.42084632873535155, "grad_norm": 1.171875, "learning_rate": 4.903887344809403e-06, "token_acc": 0.8909899302321495, "epoch": 0.10686044026501389, "step": 1000}, {"loss": 0.43503849029541014, "grad_norm": 1.390625, "learning_rate": 4.8789611462547414e-06, "token_acc": 0.8900309663514769, "epoch": 0.11754648429151528, "step": 1100}, {"loss": 0.4078071594238281, "grad_norm": 1.21875, "learning_rate": 4.851243008575364e-06, "token_acc": 0.8972086577408744, "epoch": 0.12823252831801668, "step": 1200}, {"loss": 0.4169163131713867, "grad_norm": 1.3046875, "learning_rate": 4.8207654616649035e-06, "token_acc": 0.8944712980277317, "epoch": 0.13891857234451807, "step": 1300}, {"loss": 0.41801651000976564, "grad_norm": 1.25, "learning_rate": 4.7875642738484475e-06, "token_acc": 0.8954553844138678, "epoch": 0.14960461637101946, "step": 1400}, {"loss": 0.42521297454833984, "grad_norm": 1.0390625, "learning_rate": 4.7516784099049705e-06, "token_acc": 0.8927475600806661, "epoch": 0.16029066039752085, "step": 1500}, {"loss": 0.40242053985595705, "grad_norm": 1.0625, "learning_rate": 4.713149985338428e-06, "token_acc": 0.8985869393070196, "epoch": 0.1709767044240222, "step": 1600}, {"loss": 0.39753116607666017, "grad_norm": 1.6953125, "learning_rate": 4.672024216951176e-06, "token_acc": 0.9003083728795708, "epoch": 0.1816627484505236, "step": 1700}, {"loss": 0.40670780181884764, "grad_norm": 1.328125, "learning_rate": 4.628349369777712e-06, "token_acc": 0.8985113659223496, "epoch": 0.192348792477025, "step": 1800}, {"loss": 0.4040363311767578, "grad_norm": 1.609375, "learning_rate": 4.58217670044103e-06, "token_acc": 0.898509424415663, "epoch": 0.20303483650352638, "step": 1900}, {"loss": 0.3897912216186523, "grad_norm": 1.3203125, "learning_rate": 4.533560396998055e-06, "token_acc": 0.9022499382853919, "epoch": 0.21372088053002777, "step": 2000}, {"eval_loss": 0.5564050674438477, "eval_runtime": 1628.334, "eval_samples_per_second": 20.433, "eval_steps_per_second": 1.277, "eval_token_acc": 0.8750522224756502, "epoch": 0.21372088053002777, "step": 2000}, {"loss": 0.39870925903320314, "grad_norm": 1.390625, "learning_rate": 4.482557515344757e-06, "token_acc": 0.8990360044410765, "epoch": 0.22440692455652916, "step": 2100}, {"loss": 0.38395835876464846, "grad_norm": 1.140625, "learning_rate": 4.429227912255589e-06, "token_acc": 0.902773223403757, "epoch": 0.23509296858303055, "step": 2200}, {"loss": 0.39471343994140623, "grad_norm": 0.96484375, "learning_rate": 4.3736341751358166e-06, "token_acc": 0.9010956332106893, "epoch": 0.24577901260953194, "step": 2300}, {"loss": 0.39796932220458986, "grad_norm": 1.4140625, "learning_rate": 4.315841548569193e-06, "token_acc": 0.9007656570069128, "epoch": 0.25646505663603336, "step": 2400}, {"loss": 0.3913582992553711, "grad_norm": 1.265625, "learning_rate": 4.255917857747183e-06, "token_acc": 0.9015067434468572, "epoch": 0.2671511006625347, "step": 2500}, {"loss": 0.38879886627197263, "grad_norm": 1.46875, "learning_rate": 4.193933428869592e-06, "token_acc": 0.9019667027692453, "epoch": 0.27783714468903614, "step": 2600}, {"loss": 0.3972347259521484, "grad_norm": 2.4375, "learning_rate": 4.129961006610028e-06, "token_acc": 0.9001422292689183, "epoch": 0.2885231887155375, "step": 2700}, {"loss": 0.3885000991821289, "grad_norm": 1.2578125, "learning_rate": 4.064075668743043e-06, "token_acc": 0.9027788505528588, "epoch": 0.2992092327420389, "step": 2800}, {"loss": 0.39520988464355467, "grad_norm": 1.265625, "learning_rate": 3.996354738033166e-06, "token_acc": 0.9011279097672186, "epoch": 0.3098952767685403, "step": 2900}, {"loss": 0.3871217346191406, "grad_norm": 0.921875, "learning_rate": 3.926877691489222e-06, "token_acc": 0.9022083313787118, "epoch": 0.3205813207950417, "step": 3000}, {"loss": 0.3810308074951172, "grad_norm": 1.375, "learning_rate": 3.855726067090433e-06, "token_acc": 0.9034641205430405, "epoch": 0.33126736482154306, "step": 3100}, {"loss": 0.37891117095947263, "grad_norm": 1.5546875, "learning_rate": 3.7829833680937804e-06, "token_acc": 0.9025715320536037, "epoch": 0.3419534088480444, "step": 3200}, {"loss": 0.40724864959716794, "grad_norm": 1.59375, "learning_rate": 3.7087349650349234e-06, "token_acc": 0.8990014549892049, "epoch": 0.35263945287454584, "step": 3300}, {"loss": 0.39273796081542967, "grad_norm": 1.1171875, "learning_rate": 3.6330679955376843e-06, "token_acc": 0.900933421493117, "epoch": 0.3633254969010472, "step": 3400}, {"loss": 0.3975730514526367, "grad_norm": 1.5, "learning_rate": 3.556071262049684e-06, "token_acc": 0.8998019157822584, "epoch": 0.3740115409275486, "step": 3500}, {"loss": 0.37645999908447264, "grad_norm": 2.140625, "learning_rate": 3.477835127624158e-06, "token_acc": 0.9051004274755519, "epoch": 0.38469758495405, "step": 3600}, {"loss": 0.4185586166381836, "grad_norm": 1.1796875, "learning_rate": 3.39845140987024e-06, "token_acc": 0.8963374986700084, "epoch": 0.3953836289805514, "step": 3700}, {"loss": 0.3819176483154297, "grad_norm": 1.0390625, "learning_rate": 3.318013273196187e-06, "token_acc": 0.9037970813391503, "epoch": 0.40606967300705277, "step": 3800}, {"loss": 0.4014278793334961, "grad_norm": 0.8828125, "learning_rate": 3.2366151194720163e-06, "token_acc": 0.8989897804049096, "epoch": 0.4167557170335542, "step": 3900}, {"loss": 0.3903131103515625, "grad_norm": 1.2265625, "learning_rate": 3.1543524772398464e-06, "token_acc": 0.9023779507523624, "epoch": 0.42744176106005555, "step": 4000}, {"eval_loss": 0.5559827089309692, "eval_runtime": 1625.5851, "eval_samples_per_second": 20.468, "eval_steps_per_second": 1.28, "eval_token_acc": 0.8759747254509199, "epoch": 0.42744176106005555, "step": 4000}], "memory": 38.603515625}
v0-20260123-014416/v0-20260123-194351/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v0-20260123-194351",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v0-20260123-194351/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7f3879f384c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v0-20260123-194351', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v0-20260123-194351/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000', hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/v0-20260123-194351/logging.jsonl ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_parameter_info": "PeftModelForCausalLM: 8321.1492M Params (5.9699M Trainable [0.0717%]), 0.0001M Buffers.", "last_model_checkpoint": null, "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000", "best_metric": 0.55598271, "global_step": 4029, "log_history": [{"epoch": 0.00010686044026501389, "grad_norm": 1.6953125, "learning_rate": 2.6595744680851065e-08, "loss": 0.7167500853538513, "step": 1, "token_acc": 0.8417487684729064}, {"epoch": 0.010686044026501388, "grad_norm": 1.4453125, "learning_rate": 2.6595744680851065e-06, "loss": 0.6973034637142913, "step": 100, "token_acc": 0.8467176205711577}, {"epoch": 0.021372088053002777, "grad_norm": 1.046875, "learning_rate": 4.999978873236658e-06, "loss": 0.6577812957763672, "step": 200, "token_acc": 0.8531807124137688}, {"epoch": 0.03205813207950417, "grad_norm": 1.328125, "learning_rate": 4.998159847367393e-06, "loss": 0.6022152709960937, "step": 300, "token_acc": 0.8620946875218537}, {"epoch": 0.04274417610600555, "grad_norm": 1.2421875, "learning_rate": 4.993408991262632e-06, "loss": 0.5575482177734375, "step": 400, "token_acc": 0.8680139909957781}, {"epoch": 0.05343022013250694, "grad_norm": 1.8984375, "learning_rate": 4.985731880507775e-06, "loss": 0.5264178466796875, "step": 500, "token_acc": 0.8696029378557931}, {"epoch": 0.06411626415900834, "grad_norm": 1.1171875, "learning_rate": 4.975137524928774e-06, "loss": 0.4771532440185547, "step": 600, "token_acc": 0.8748165713816137}, {"epoch": 0.07480230818550973, "grad_norm": 1.4765625, "learning_rate": 4.961638358018247e-06, "loss": 0.4643104553222656, "step": 700, "token_acc": 0.8754063671076119}, {"epoch": 0.0854883522120111, "grad_norm": 1.109375, "learning_rate": 4.945250222343571e-06, "loss": 0.4561585235595703, "step": 800, "token_acc": 0.8795579017210171}, {"epoch": 0.0961743962385125, "grad_norm": 1.171875, "learning_rate": 4.925992350954112e-06, "loss": 0.4534299850463867, "step": 900, "token_acc": 0.8851879828938838}, {"epoch": 0.10686044026501389, "grad_norm": 1.171875, "learning_rate": 4.903887344809403e-06, "loss": 0.42084632873535155, "step": 1000, "token_acc": 0.8909899302321495}, {"epoch": 0.11754648429151528, "grad_norm": 1.390625, "learning_rate": 4.8789611462547414e-06, "loss": 0.43503849029541014, "step": 1100, "token_acc": 0.8900309663514769}, {"epoch": 0.12823252831801668, "grad_norm": 1.21875, "learning_rate": 4.851243008575364e-06, "loss": 0.4078071594238281, "step": 1200, "token_acc": 0.8972086577408744}, {"epoch": 0.13891857234451807, "grad_norm": 1.3046875, "learning_rate": 4.8207654616649035e-06, "loss": 0.4169163131713867, "step": 1300, "token_acc": 0.8944712980277317}, {"epoch": 0.14960461637101946, "grad_norm": 1.25, "learning_rate": 4.7875642738484475e-06, "loss": 0.41801651000976564, "step": 1400, "token_acc": 0.8954553844138678}, {"epoch": 0.16029066039752085, "grad_norm": 1.0390625, "learning_rate": 4.7516784099049705e-06, "loss": 0.42521297454833984, "step": 1500, "token_acc": 0.8927475600806661}, {"epoch": 0.1709767044240222, "grad_norm": 1.0625, "learning_rate": 4.713149985338428e-06, "loss": 0.40242053985595705, "step": 1600, "token_acc": 0.8985869393070196}, {"epoch": 0.1816627484505236, "grad_norm": 1.6953125, "learning_rate": 4.672024216951176e-06, "loss": 0.39753116607666017, "step": 1700, "token_acc": 0.9003083728795708}, {"epoch": 0.192348792477025, "grad_norm": 1.328125, "learning_rate": 4.628349369777712e-06, "loss": 0.40670780181884764, "step": 1800, "token_acc": 0.8985113659223496}, {"epoch": 0.20303483650352638, "grad_norm": 1.609375, "learning_rate": 4.58217670044103e-06, "loss": 0.4040363311767578, "step": 1900, "token_acc": 0.898509424415663}, {"epoch": 0.21372088053002777, "grad_norm": 1.3203125, "learning_rate": 4.533560396998055e-06, "loss": 0.3897912216186523, "step": 2000, "token_acc": 0.9022499382853919}, {"epoch": 0.21372088053002777, "eval_loss": 0.5564050674438477, "eval_runtime": 1628.334, "eval_samples_per_second": 20.433, "eval_steps_per_second": 1.277, "eval_token_acc": 0.8750522224756502, "step": 2000}, {"epoch": 0.22440692455652916, "grad_norm": 1.390625, "learning_rate": 4.482557515344757e-06, "loss": 0.39870925903320314, "step": 2100, "token_acc": 0.8990360044410765}, {"epoch": 0.23509296858303055, "grad_norm": 1.140625, "learning_rate": 4.429227912255589e-06, "loss": 0.38395835876464846, "step": 2200, "token_acc": 0.902773223403757}, {"epoch": 0.24577901260953194, "grad_norm": 0.96484375, "learning_rate": 4.3736341751358166e-06, "loss": 0.39471343994140623, "step": 2300, "token_acc": 0.9010956332106893}, {"epoch": 0.25646505663603336, "grad_norm": 1.4140625, "learning_rate": 4.315841548569193e-06, "loss": 0.39796932220458986, "step": 2400, "token_acc": 0.9007656570069128}, {"epoch": 0.2671511006625347, "grad_norm": 1.265625, "learning_rate": 4.255917857747183e-06, "loss": 0.3913582992553711, "step": 2500, "token_acc": 0.9015067434468572}, {"epoch": 0.27783714468903614, "grad_norm": 1.46875, "learning_rate": 4.193933428869592e-06, "loss": 0.38879886627197263, "step": 2600, "token_acc": 0.9019667027692453}, {"epoch": 0.2885231887155375, "grad_norm": 2.4375, "learning_rate": 4.129961006610028e-06, "loss": 0.3972347259521484, "step": 2700, "token_acc": 0.9001422292689183}, {"epoch": 0.2992092327420389, "grad_norm": 1.2578125, "learning_rate": 4.064075668743043e-06, "loss": 0.3885000991821289, "step": 2800, "token_acc": 0.9027788505528588}, {"epoch": 0.3098952767685403, "grad_norm": 1.265625, "learning_rate": 3.996354738033166e-06, "loss": 0.39520988464355467, "step": 2900, "token_acc": 0.9011279097672186}, {"epoch": 0.3205813207950417, "grad_norm": 0.921875, "learning_rate": 3.926877691489222e-06, "loss": 0.3871217346191406, "step": 3000, "token_acc": 0.9022083313787118}, {"epoch": 0.33126736482154306, "grad_norm": 1.375, "learning_rate": 3.855726067090433e-06, "loss": 0.3810308074951172, "step": 3100, "token_acc": 0.9034641205430405}, {"epoch": 0.3419534088480444, "grad_norm": 1.5546875, "learning_rate": 3.7829833680937804e-06, "loss": 0.37891117095947263, "step": 3200, "token_acc": 0.9025715320536037}, {"epoch": 0.35263945287454584, "grad_norm": 1.59375, "learning_rate": 3.7087349650349234e-06, "loss": 0.40724864959716794, "step": 3300, "token_acc": 0.8990014549892049}, {"epoch": 0.3633254969010472, "grad_norm": 1.1171875, "learning_rate": 3.6330679955376843e-06, "loss": 0.39273796081542967, "step": 3400, "token_acc": 0.900933421493117}, {"epoch": 0.3740115409275486, "grad_norm": 1.5, "learning_rate": 3.556071262049684e-06, "loss": 0.3975730514526367, "step": 3500, "token_acc": 0.8998019157822584}, {"epoch": 0.38469758495405, "grad_norm": 2.140625, "learning_rate": 3.477835127624158e-06, "loss": 0.37645999908447264, "step": 3600, "token_acc": 0.9051004274755519}, {"epoch": 0.3953836289805514, "grad_norm": 1.1796875, "learning_rate": 3.39845140987024e-06, "loss": 0.4185586166381836, "step": 3700, "token_acc": 0.8963374986700084}, {"epoch": 0.40606967300705277, "grad_norm": 1.0390625, "learning_rate": 3.318013273196187e-06, "loss": 0.3819176483154297, "step": 3800, "token_acc": 0.9037970813391503}, {"epoch": 0.4167557170335542, "grad_norm": 0.8828125, "learning_rate": 3.2366151194720163e-06, "loss": 0.4014278793334961, "step": 3900, "token_acc": 0.8989897804049096}, {"epoch": 0.42744176106005555, "grad_norm": 1.2265625, "learning_rate": 3.1543524772398464e-06, "loss": 0.3903131103515625, "step": 4000, "token_acc": 0.9023779507523624}, {"epoch": 0.42744176106005555, "eval_loss": 0.5559827089309692, "eval_runtime": 1625.5851, "eval_samples_per_second": 20.468, "eval_steps_per_second": 1.28, "eval_token_acc": 0.8759747254509199, "step": 4000}], "memory": null}
v0-20260123-014416/v1-20260123-200453/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7fdf2b1c04c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000', hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [],
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": "^(encoder.*\\.(query|mlp.0|mlp.2|out|value|key)|adapter.*\\.(linear1|linear2))$",
23
+ "task_type": "CAUSAL_LM",
24
+ "use_dora": false,
25
+ "use_rslora": false
26
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50a0e1eda4ba03132030371f0fa7d66bbda63e1cb52788ff97b33a49dad31151
3
+ size 11988912
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": null, "lorap_emb_lr": 1e-06}
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7fdf2b1c04c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000', hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc9b96b78c9730edc2d4dc127460baa4f6a329480bcffd1918a9f1a51943afb5
3
+ size 24205999
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31ab0803353fc2e7510dbdc9e623d0a5732ec637183db5671f27b071bfc5e3bc
3
+ size 14645
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:753710a5ff55912e7374c1a6e5097327041812bfc690b36bba9fc912adb2d1db
3
+ size 1465
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/trainer_state.json ADDED
@@ -0,0 +1,548 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.55598271,
3
+ "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
4
+ "epoch": 0.6411626415900834,
5
+ "eval_steps": 2000,
6
+ "global_step": 6000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.00010686044026501389,
13
+ "grad_norm": 1.6953125,
14
+ "learning_rate": 2.6595744680851065e-08,
15
+ "loss": 0.7167500853538513,
16
+ "step": 1,
17
+ "token_acc": 0.8417487684729064
18
+ },
19
+ {
20
+ "epoch": 0.010686044026501388,
21
+ "grad_norm": 1.4453125,
22
+ "learning_rate": 2.6595744680851065e-06,
23
+ "loss": 0.6973034637142913,
24
+ "step": 100,
25
+ "token_acc": 0.8467176205711577
26
+ },
27
+ {
28
+ "epoch": 0.021372088053002777,
29
+ "grad_norm": 1.046875,
30
+ "learning_rate": 4.999978873236658e-06,
31
+ "loss": 0.6577812957763672,
32
+ "step": 200,
33
+ "token_acc": 0.8531807124137688
34
+ },
35
+ {
36
+ "epoch": 0.03205813207950417,
37
+ "grad_norm": 1.328125,
38
+ "learning_rate": 4.998159847367393e-06,
39
+ "loss": 0.6022152709960937,
40
+ "step": 300,
41
+ "token_acc": 0.8620946875218537
42
+ },
43
+ {
44
+ "epoch": 0.04274417610600555,
45
+ "grad_norm": 1.2421875,
46
+ "learning_rate": 4.993408991262632e-06,
47
+ "loss": 0.5575482177734375,
48
+ "step": 400,
49
+ "token_acc": 0.8680139909957781
50
+ },
51
+ {
52
+ "epoch": 0.05343022013250694,
53
+ "grad_norm": 1.8984375,
54
+ "learning_rate": 4.985731880507775e-06,
55
+ "loss": 0.5264178466796875,
56
+ "step": 500,
57
+ "token_acc": 0.8696029378557931
58
+ },
59
+ {
60
+ "epoch": 0.06411626415900834,
61
+ "grad_norm": 1.1171875,
62
+ "learning_rate": 4.975137524928774e-06,
63
+ "loss": 0.4771532440185547,
64
+ "step": 600,
65
+ "token_acc": 0.8748165713816137
66
+ },
67
+ {
68
+ "epoch": 0.07480230818550973,
69
+ "grad_norm": 1.4765625,
70
+ "learning_rate": 4.961638358018247e-06,
71
+ "loss": 0.4643104553222656,
72
+ "step": 700,
73
+ "token_acc": 0.8754063671076119
74
+ },
75
+ {
76
+ "epoch": 0.0854883522120111,
77
+ "grad_norm": 1.109375,
78
+ "learning_rate": 4.945250222343571e-06,
79
+ "loss": 0.4561585235595703,
80
+ "step": 800,
81
+ "token_acc": 0.8795579017210171
82
+ },
83
+ {
84
+ "epoch": 0.0961743962385125,
85
+ "grad_norm": 1.171875,
86
+ "learning_rate": 4.925992350954112e-06,
87
+ "loss": 0.4534299850463867,
88
+ "step": 900,
89
+ "token_acc": 0.8851879828938838
90
+ },
91
+ {
92
+ "epoch": 0.10686044026501389,
93
+ "grad_norm": 1.171875,
94
+ "learning_rate": 4.903887344809403e-06,
95
+ "loss": 0.42084632873535155,
96
+ "step": 1000,
97
+ "token_acc": 0.8909899302321495
98
+ },
99
+ {
100
+ "epoch": 0.11754648429151528,
101
+ "grad_norm": 1.390625,
102
+ "learning_rate": 4.8789611462547414e-06,
103
+ "loss": 0.43503849029541014,
104
+ "step": 1100,
105
+ "token_acc": 0.8900309663514769
106
+ },
107
+ {
108
+ "epoch": 0.12823252831801668,
109
+ "grad_norm": 1.21875,
110
+ "learning_rate": 4.851243008575364e-06,
111
+ "loss": 0.4078071594238281,
112
+ "step": 1200,
113
+ "token_acc": 0.8972086577408744
114
+ },
115
+ {
116
+ "epoch": 0.13891857234451807,
117
+ "grad_norm": 1.3046875,
118
+ "learning_rate": 4.8207654616649035e-06,
119
+ "loss": 0.4169163131713867,
120
+ "step": 1300,
121
+ "token_acc": 0.8944712980277317
122
+ },
123
+ {
124
+ "epoch": 0.14960461637101946,
125
+ "grad_norm": 1.25,
126
+ "learning_rate": 4.7875642738484475e-06,
127
+ "loss": 0.41801651000976564,
128
+ "step": 1400,
129
+ "token_acc": 0.8954553844138678
130
+ },
131
+ {
132
+ "epoch": 0.16029066039752085,
133
+ "grad_norm": 1.0390625,
134
+ "learning_rate": 4.7516784099049705e-06,
135
+ "loss": 0.42521297454833984,
136
+ "step": 1500,
137
+ "token_acc": 0.8927475600806661
138
+ },
139
+ {
140
+ "epoch": 0.1709767044240222,
141
+ "grad_norm": 1.0625,
142
+ "learning_rate": 4.713149985338428e-06,
143
+ "loss": 0.40242053985595705,
144
+ "step": 1600,
145
+ "token_acc": 0.8985869393070196
146
+ },
147
+ {
148
+ "epoch": 0.1816627484505236,
149
+ "grad_norm": 1.6953125,
150
+ "learning_rate": 4.672024216951176e-06,
151
+ "loss": 0.39753116607666017,
152
+ "step": 1700,
153
+ "token_acc": 0.9003083728795708
154
+ },
155
+ {
156
+ "epoch": 0.192348792477025,
157
+ "grad_norm": 1.328125,
158
+ "learning_rate": 4.628349369777712e-06,
159
+ "loss": 0.40670780181884764,
160
+ "step": 1800,
161
+ "token_acc": 0.8985113659223496
162
+ },
163
+ {
164
+ "epoch": 0.20303483650352638,
165
+ "grad_norm": 1.609375,
166
+ "learning_rate": 4.58217670044103e-06,
167
+ "loss": 0.4040363311767578,
168
+ "step": 1900,
169
+ "token_acc": 0.898509424415663
170
+ },
171
+ {
172
+ "epoch": 0.21372088053002777,
173
+ "grad_norm": 1.3203125,
174
+ "learning_rate": 4.533560396998055e-06,
175
+ "loss": 0.3897912216186523,
176
+ "step": 2000,
177
+ "token_acc": 0.9022499382853919
178
+ },
179
+ {
180
+ "epoch": 0.21372088053002777,
181
+ "eval_loss": 0.5564050674438477,
182
+ "eval_runtime": 1628.334,
183
+ "eval_samples_per_second": 20.433,
184
+ "eval_steps_per_second": 1.277,
185
+ "eval_token_acc": 0.8750522224756502,
186
+ "step": 2000
187
+ },
188
+ {
189
+ "epoch": 0.22440692455652916,
190
+ "grad_norm": 1.390625,
191
+ "learning_rate": 4.482557515344757e-06,
192
+ "loss": 0.39870925903320314,
193
+ "step": 2100,
194
+ "token_acc": 0.8990360044410765
195
+ },
196
+ {
197
+ "epoch": 0.23509296858303055,
198
+ "grad_norm": 1.140625,
199
+ "learning_rate": 4.429227912255589e-06,
200
+ "loss": 0.38395835876464846,
201
+ "step": 2200,
202
+ "token_acc": 0.902773223403757
203
+ },
204
+ {
205
+ "epoch": 0.24577901260953194,
206
+ "grad_norm": 0.96484375,
207
+ "learning_rate": 4.3736341751358166e-06,
208
+ "loss": 0.39471343994140623,
209
+ "step": 2300,
210
+ "token_acc": 0.9010956332106893
211
+ },
212
+ {
213
+ "epoch": 0.25646505663603336,
214
+ "grad_norm": 1.4140625,
215
+ "learning_rate": 4.315841548569193e-06,
216
+ "loss": 0.39796932220458986,
217
+ "step": 2400,
218
+ "token_acc": 0.9007656570069128
219
+ },
220
+ {
221
+ "epoch": 0.2671511006625347,
222
+ "grad_norm": 1.265625,
223
+ "learning_rate": 4.255917857747183e-06,
224
+ "loss": 0.3913582992553711,
225
+ "step": 2500,
226
+ "token_acc": 0.9015067434468572
227
+ },
228
+ {
229
+ "epoch": 0.27783714468903614,
230
+ "grad_norm": 1.46875,
231
+ "learning_rate": 4.193933428869592e-06,
232
+ "loss": 0.38879886627197263,
233
+ "step": 2600,
234
+ "token_acc": 0.9019667027692453
235
+ },
236
+ {
237
+ "epoch": 0.2885231887155375,
238
+ "grad_norm": 2.4375,
239
+ "learning_rate": 4.129961006610028e-06,
240
+ "loss": 0.3972347259521484,
241
+ "step": 2700,
242
+ "token_acc": 0.9001422292689183
243
+ },
244
+ {
245
+ "epoch": 0.2992092327420389,
246
+ "grad_norm": 1.2578125,
247
+ "learning_rate": 4.064075668743043e-06,
248
+ "loss": 0.3885000991821289,
249
+ "step": 2800,
250
+ "token_acc": 0.9027788505528588
251
+ },
252
+ {
253
+ "epoch": 0.3098952767685403,
254
+ "grad_norm": 1.265625,
255
+ "learning_rate": 3.996354738033166e-06,
256
+ "loss": 0.39520988464355467,
257
+ "step": 2900,
258
+ "token_acc": 0.9011279097672186
259
+ },
260
+ {
261
+ "epoch": 0.3205813207950417,
262
+ "grad_norm": 0.921875,
263
+ "learning_rate": 3.926877691489222e-06,
264
+ "loss": 0.3871217346191406,
265
+ "step": 3000,
266
+ "token_acc": 0.9022083313787118
267
+ },
268
+ {
269
+ "epoch": 0.33126736482154306,
270
+ "grad_norm": 1.375,
271
+ "learning_rate": 3.855726067090433e-06,
272
+ "loss": 0.3810308074951172,
273
+ "step": 3100,
274
+ "token_acc": 0.9034641205430405
275
+ },
276
+ {
277
+ "epoch": 0.3419534088480444,
278
+ "grad_norm": 1.5546875,
279
+ "learning_rate": 3.7829833680937804e-06,
280
+ "loss": 0.37891117095947263,
281
+ "step": 3200,
282
+ "token_acc": 0.9025715320536037
283
+ },
284
+ {
285
+ "epoch": 0.35263945287454584,
286
+ "grad_norm": 1.59375,
287
+ "learning_rate": 3.7087349650349234e-06,
288
+ "loss": 0.40724864959716794,
289
+ "step": 3300,
290
+ "token_acc": 0.8990014549892049
291
+ },
292
+ {
293
+ "epoch": 0.3633254969010472,
294
+ "grad_norm": 1.1171875,
295
+ "learning_rate": 3.6330679955376843e-06,
296
+ "loss": 0.39273796081542967,
297
+ "step": 3400,
298
+ "token_acc": 0.900933421493117
299
+ },
300
+ {
301
+ "epoch": 0.3740115409275486,
302
+ "grad_norm": 1.5,
303
+ "learning_rate": 3.556071262049684e-06,
304
+ "loss": 0.3975730514526367,
305
+ "step": 3500,
306
+ "token_acc": 0.8998019157822584
307
+ },
308
+ {
309
+ "epoch": 0.38469758495405,
310
+ "grad_norm": 2.140625,
311
+ "learning_rate": 3.477835127624158e-06,
312
+ "loss": 0.37645999908447264,
313
+ "step": 3600,
314
+ "token_acc": 0.9051004274755519
315
+ },
316
+ {
317
+ "epoch": 0.3953836289805514,
318
+ "grad_norm": 1.1796875,
319
+ "learning_rate": 3.39845140987024e-06,
320
+ "loss": 0.4185586166381836,
321
+ "step": 3700,
322
+ "token_acc": 0.8963374986700084
323
+ },
324
+ {
325
+ "epoch": 0.40606967300705277,
326
+ "grad_norm": 1.0390625,
327
+ "learning_rate": 3.318013273196187e-06,
328
+ "loss": 0.3819176483154297,
329
+ "step": 3800,
330
+ "token_acc": 0.9037970813391503
331
+ },
332
+ {
333
+ "epoch": 0.4167557170335542,
334
+ "grad_norm": 0.8828125,
335
+ "learning_rate": 3.2366151194720163e-06,
336
+ "loss": 0.4014278793334961,
337
+ "step": 3900,
338
+ "token_acc": 0.8989897804049096
339
+ },
340
+ {
341
+ "epoch": 0.42744176106005555,
342
+ "grad_norm": 1.2265625,
343
+ "learning_rate": 3.1543524772398464e-06,
344
+ "loss": 0.3903131103515625,
345
+ "step": 4000,
346
+ "token_acc": 0.9023779507523624
347
+ },
348
+ {
349
+ "epoch": 0.42744176106005555,
350
+ "eval_loss": 0.5559827089309692,
351
+ "eval_runtime": 1625.5851,
352
+ "eval_samples_per_second": 20.468,
353
+ "eval_steps_per_second": 1.28,
354
+ "eval_token_acc": 0.8759747254509199,
355
+ "step": 4000
356
+ },
357
+ {
358
+ "epoch": 0.43812780508655697,
359
+ "grad_norm": 1.5078125,
360
+ "learning_rate": 3.071321889601996e-06,
361
+ "loss": 0.3918019866943359,
362
+ "step": 4100,
363
+ "token_acc": 0.9021251892681234
364
+ },
365
+ {
366
+ "epoch": 0.4488138491130583,
367
+ "grad_norm": 1.3125,
368
+ "learning_rate": 2.9876208009183837e-06,
369
+ "loss": 0.39420204162597655,
370
+ "step": 4200,
371
+ "token_acc": 0.9012835119864208
372
+ },
373
+ {
374
+ "epoch": 0.45949989313955975,
375
+ "grad_norm": 3.140625,
376
+ "learning_rate": 2.9033474424462336e-06,
377
+ "loss": 0.39763076782226564,
378
+ "step": 4300,
379
+ "token_acc": 0.9007918217810081
380
+ },
381
+ {
382
+ "epoch": 0.4701859371660611,
383
+ "grad_norm": 1.1328125,
384
+ "learning_rate": 2.8186007170562607e-06,
385
+ "loss": 0.3739194869995117,
386
+ "step": 4400,
387
+ "token_acc": 0.9055805296059564
388
+ },
389
+ {
390
+ "epoch": 0.4808719811925625,
391
+ "grad_norm": 1.015625,
392
+ "learning_rate": 2.733480083160674e-06,
393
+ "loss": 0.3924733734130859,
394
+ "step": 4500,
395
+ "token_acc": 0.9005020191807561
396
+ },
397
+ {
398
+ "epoch": 0.4915580252190639,
399
+ "grad_norm": 1.140625,
400
+ "learning_rate": 2.648085437989179e-06,
401
+ "loss": 0.3818220520019531,
402
+ "step": 4600,
403
+ "token_acc": 0.9031686537753115
404
+ },
405
+ {
406
+ "epoch": 0.5022440692455653,
407
+ "grad_norm": 1.1875,
408
+ "learning_rate": 2.5625170003500053e-06,
409
+ "loss": 0.3947904968261719,
410
+ "step": 4700,
411
+ "token_acc": 0.9003777792274801
412
+ },
413
+ {
414
+ "epoch": 0.5129301132720667,
415
+ "grad_norm": 0.98046875,
416
+ "learning_rate": 2.4768751930135102e-06,
417
+ "loss": 0.39525905609130857,
418
+ "step": 4800,
419
+ "token_acc": 0.9012966596534361
420
+ },
421
+ {
422
+ "epoch": 0.523616157298568,
423
+ "grad_norm": 0.8984375,
424
+ "learning_rate": 2.391260524856434e-06,
425
+ "loss": 0.3915714645385742,
426
+ "step": 4900,
427
+ "token_acc": 0.9022641509433962
428
+ },
429
+ {
430
+ "epoch": 0.5343022013250694,
431
+ "grad_norm": 1.3671875,
432
+ "learning_rate": 2.3057734729050834e-06,
433
+ "loss": 0.3926304626464844,
434
+ "step": 5000,
435
+ "token_acc": 0.9013136721378624
436
+ },
437
+ {
438
+ "epoch": 0.5449882453515709,
439
+ "grad_norm": 1.34375,
440
+ "learning_rate": 2.2205143644159065e-06,
441
+ "loss": 0.3773085403442383,
442
+ "step": 5100,
443
+ "token_acc": 0.9033588372573927
444
+ },
445
+ {
446
+ "epoch": 0.5556742893780723,
447
+ "grad_norm": 1.109375,
448
+ "learning_rate": 2.135583259131831e-06,
449
+ "loss": 0.39842689514160157,
450
+ "step": 5200,
451
+ "token_acc": 0.9005042691639996
452
+ },
453
+ {
454
+ "epoch": 0.5663603334045736,
455
+ "grad_norm": 0.890625,
456
+ "learning_rate": 2.0510798318525585e-06,
457
+ "loss": 0.39046684265136716,
458
+ "step": 5300,
459
+ "token_acc": 0.9020825721592985
460
+ },
461
+ {
462
+ "epoch": 0.577046377431075,
463
+ "grad_norm": 0.92578125,
464
+ "learning_rate": 1.9671032554566205e-06,
465
+ "loss": 0.39618133544921874,
466
+ "step": 5400,
467
+ "token_acc": 0.9000712137530825
468
+ },
469
+ {
470
+ "epoch": 0.5877324214575764,
471
+ "grad_norm": 0.96484375,
472
+ "learning_rate": 1.8837520845124985e-06,
473
+ "loss": 0.3802578353881836,
474
+ "step": 5500,
475
+ "token_acc": 0.9039277912871451
476
+ },
477
+ {
478
+ "epoch": 0.5984184654840778,
479
+ "grad_norm": 2.265625,
480
+ "learning_rate": 1.8011241396153775e-06,
481
+ "loss": 0.40635391235351564,
482
+ "step": 5600,
483
+ "token_acc": 0.8981563749444691
484
+ },
485
+ {
486
+ "epoch": 0.6091045095105792,
487
+ "grad_norm": 1.53125,
488
+ "learning_rate": 1.7193163925852968e-06,
489
+ "loss": 0.36938583374023437,
490
+ "step": 5700,
491
+ "token_acc": 0.9048525545411809
492
+ },
493
+ {
494
+ "epoch": 0.6197905535370806,
495
+ "grad_norm": 1.0703125,
496
+ "learning_rate": 1.6384248526614122e-06,
497
+ "loss": 0.37309234619140624,
498
+ "step": 5800,
499
+ "token_acc": 0.9052111091372061
500
+ },
501
+ {
502
+ "epoch": 0.630476597563582,
503
+ "grad_norm": 1.515625,
504
+ "learning_rate": 1.558544453825951e-06,
505
+ "loss": 0.3821083068847656,
506
+ "step": 5900,
507
+ "token_acc": 0.9030327643665118
508
+ },
509
+ {
510
+ "epoch": 0.6411626415900834,
511
+ "grad_norm": 1.0390625,
512
+ "learning_rate": 1.479768943390068e-06,
513
+ "loss": 0.3839926910400391,
514
+ "step": 6000,
515
+ "token_acc": 0.9032827727270024
516
+ },
517
+ {
518
+ "epoch": 0.6411626415900834,
519
+ "eval_loss": 0.5562668442726135,
520
+ "eval_runtime": 1881.2991,
521
+ "eval_samples_per_second": 17.686,
522
+ "eval_steps_per_second": 1.106,
523
+ "eval_token_acc": 0.8759600467010075,
524
+ "step": 6000
525
+ }
526
+ ],
527
+ "logging_steps": 100,
528
+ "max_steps": 9358,
529
+ "num_input_tokens_seen": 0,
530
+ "num_train_epochs": 1,
531
+ "save_steps": 2000,
532
+ "stateful_callbacks": {
533
+ "TrainerControl": {
534
+ "args": {
535
+ "should_epoch_stop": false,
536
+ "should_evaluate": false,
537
+ "should_log": false,
538
+ "should_save": true,
539
+ "should_training_stop": false
540
+ },
541
+ "attributes": {}
542
+ }
543
+ },
544
+ "total_flos": 4.827843079406936e+18,
545
+ "train_batch_size": 16,
546
+ "trial_name": null,
547
+ "trial_params": null
548
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-6000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:859f9e6349e507974dc4ff76def0df20e49d08364f05bd0551845ae28b369820
3
+ size 7377
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [],
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": "^(encoder.*\\.(query|mlp.0|mlp.2|out|value|key)|adapter.*\\.(linear1|linear2))$",
23
+ "task_type": "CAUSAL_LM",
24
+ "use_dora": false,
25
+ "use_rslora": false
26
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbf77831dbe7d86e3688535db09f8cf5423a04343f5e3fb26a5ff6a7456da1e7
3
+ size 11988912
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": null, "lorap_emb_lr": 1e-06}
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7fdf2b1c04c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000', hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eaa54ce4d6f12732345e3b95c9ee67831a6193ad74077acda3f2ba955eb265fc
3
+ size 24205999
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89ffad4d855b6ed34e153b28369a58961a10b7b742b35f86bf7be3f0a44a3466
3
+ size 14645
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:770c50c069dfed51cddb935a91c450607285f8f9a9c450d76c7abaf90044540f
3
+ size 1465
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/trainer_state.json ADDED
@@ -0,0 +1,717 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.55598271,
3
+ "best_model_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
4
+ "epoch": 0.8548835221201111,
5
+ "eval_steps": 2000,
6
+ "global_step": 8000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.00010686044026501389,
13
+ "grad_norm": 1.6953125,
14
+ "learning_rate": 2.6595744680851065e-08,
15
+ "loss": 0.7167500853538513,
16
+ "step": 1,
17
+ "token_acc": 0.8417487684729064
18
+ },
19
+ {
20
+ "epoch": 0.010686044026501388,
21
+ "grad_norm": 1.4453125,
22
+ "learning_rate": 2.6595744680851065e-06,
23
+ "loss": 0.6973034637142913,
24
+ "step": 100,
25
+ "token_acc": 0.8467176205711577
26
+ },
27
+ {
28
+ "epoch": 0.021372088053002777,
29
+ "grad_norm": 1.046875,
30
+ "learning_rate": 4.999978873236658e-06,
31
+ "loss": 0.6577812957763672,
32
+ "step": 200,
33
+ "token_acc": 0.8531807124137688
34
+ },
35
+ {
36
+ "epoch": 0.03205813207950417,
37
+ "grad_norm": 1.328125,
38
+ "learning_rate": 4.998159847367393e-06,
39
+ "loss": 0.6022152709960937,
40
+ "step": 300,
41
+ "token_acc": 0.8620946875218537
42
+ },
43
+ {
44
+ "epoch": 0.04274417610600555,
45
+ "grad_norm": 1.2421875,
46
+ "learning_rate": 4.993408991262632e-06,
47
+ "loss": 0.5575482177734375,
48
+ "step": 400,
49
+ "token_acc": 0.8680139909957781
50
+ },
51
+ {
52
+ "epoch": 0.05343022013250694,
53
+ "grad_norm": 1.8984375,
54
+ "learning_rate": 4.985731880507775e-06,
55
+ "loss": 0.5264178466796875,
56
+ "step": 500,
57
+ "token_acc": 0.8696029378557931
58
+ },
59
+ {
60
+ "epoch": 0.06411626415900834,
61
+ "grad_norm": 1.1171875,
62
+ "learning_rate": 4.975137524928774e-06,
63
+ "loss": 0.4771532440185547,
64
+ "step": 600,
65
+ "token_acc": 0.8748165713816137
66
+ },
67
+ {
68
+ "epoch": 0.07480230818550973,
69
+ "grad_norm": 1.4765625,
70
+ "learning_rate": 4.961638358018247e-06,
71
+ "loss": 0.4643104553222656,
72
+ "step": 700,
73
+ "token_acc": 0.8754063671076119
74
+ },
75
+ {
76
+ "epoch": 0.0854883522120111,
77
+ "grad_norm": 1.109375,
78
+ "learning_rate": 4.945250222343571e-06,
79
+ "loss": 0.4561585235595703,
80
+ "step": 800,
81
+ "token_acc": 0.8795579017210171
82
+ },
83
+ {
84
+ "epoch": 0.0961743962385125,
85
+ "grad_norm": 1.171875,
86
+ "learning_rate": 4.925992350954112e-06,
87
+ "loss": 0.4534299850463867,
88
+ "step": 900,
89
+ "token_acc": 0.8851879828938838
90
+ },
91
+ {
92
+ "epoch": 0.10686044026501389,
93
+ "grad_norm": 1.171875,
94
+ "learning_rate": 4.903887344809403e-06,
95
+ "loss": 0.42084632873535155,
96
+ "step": 1000,
97
+ "token_acc": 0.8909899302321495
98
+ },
99
+ {
100
+ "epoch": 0.11754648429151528,
101
+ "grad_norm": 1.390625,
102
+ "learning_rate": 4.8789611462547414e-06,
103
+ "loss": 0.43503849029541014,
104
+ "step": 1100,
105
+ "token_acc": 0.8900309663514769
106
+ },
107
+ {
108
+ "epoch": 0.12823252831801668,
109
+ "grad_norm": 1.21875,
110
+ "learning_rate": 4.851243008575364e-06,
111
+ "loss": 0.4078071594238281,
112
+ "step": 1200,
113
+ "token_acc": 0.8972086577408744
114
+ },
115
+ {
116
+ "epoch": 0.13891857234451807,
117
+ "grad_norm": 1.3046875,
118
+ "learning_rate": 4.8207654616649035e-06,
119
+ "loss": 0.4169163131713867,
120
+ "step": 1300,
121
+ "token_acc": 0.8944712980277317
122
+ },
123
+ {
124
+ "epoch": 0.14960461637101946,
125
+ "grad_norm": 1.25,
126
+ "learning_rate": 4.7875642738484475e-06,
127
+ "loss": 0.41801651000976564,
128
+ "step": 1400,
129
+ "token_acc": 0.8954553844138678
130
+ },
131
+ {
132
+ "epoch": 0.16029066039752085,
133
+ "grad_norm": 1.0390625,
134
+ "learning_rate": 4.7516784099049705e-06,
135
+ "loss": 0.42521297454833984,
136
+ "step": 1500,
137
+ "token_acc": 0.8927475600806661
138
+ },
139
+ {
140
+ "epoch": 0.1709767044240222,
141
+ "grad_norm": 1.0625,
142
+ "learning_rate": 4.713149985338428e-06,
143
+ "loss": 0.40242053985595705,
144
+ "step": 1600,
145
+ "token_acc": 0.8985869393070196
146
+ },
147
+ {
148
+ "epoch": 0.1816627484505236,
149
+ "grad_norm": 1.6953125,
150
+ "learning_rate": 4.672024216951176e-06,
151
+ "loss": 0.39753116607666017,
152
+ "step": 1700,
153
+ "token_acc": 0.9003083728795708
154
+ },
155
+ {
156
+ "epoch": 0.192348792477025,
157
+ "grad_norm": 1.328125,
158
+ "learning_rate": 4.628349369777712e-06,
159
+ "loss": 0.40670780181884764,
160
+ "step": 1800,
161
+ "token_acc": 0.8985113659223496
162
+ },
163
+ {
164
+ "epoch": 0.20303483650352638,
165
+ "grad_norm": 1.609375,
166
+ "learning_rate": 4.58217670044103e-06,
167
+ "loss": 0.4040363311767578,
168
+ "step": 1900,
169
+ "token_acc": 0.898509424415663
170
+ },
171
+ {
172
+ "epoch": 0.21372088053002777,
173
+ "grad_norm": 1.3203125,
174
+ "learning_rate": 4.533560396998055e-06,
175
+ "loss": 0.3897912216186523,
176
+ "step": 2000,
177
+ "token_acc": 0.9022499382853919
178
+ },
179
+ {
180
+ "epoch": 0.21372088053002777,
181
+ "eval_loss": 0.5564050674438477,
182
+ "eval_runtime": 1628.334,
183
+ "eval_samples_per_second": 20.433,
184
+ "eval_steps_per_second": 1.277,
185
+ "eval_token_acc": 0.8750522224756502,
186
+ "step": 2000
187
+ },
188
+ {
189
+ "epoch": 0.22440692455652916,
190
+ "grad_norm": 1.390625,
191
+ "learning_rate": 4.482557515344757e-06,
192
+ "loss": 0.39870925903320314,
193
+ "step": 2100,
194
+ "token_acc": 0.8990360044410765
195
+ },
196
+ {
197
+ "epoch": 0.23509296858303055,
198
+ "grad_norm": 1.140625,
199
+ "learning_rate": 4.429227912255589e-06,
200
+ "loss": 0.38395835876464846,
201
+ "step": 2200,
202
+ "token_acc": 0.902773223403757
203
+ },
204
+ {
205
+ "epoch": 0.24577901260953194,
206
+ "grad_norm": 0.96484375,
207
+ "learning_rate": 4.3736341751358166e-06,
208
+ "loss": 0.39471343994140623,
209
+ "step": 2300,
210
+ "token_acc": 0.9010956332106893
211
+ },
212
+ {
213
+ "epoch": 0.25646505663603336,
214
+ "grad_norm": 1.4140625,
215
+ "learning_rate": 4.315841548569193e-06,
216
+ "loss": 0.39796932220458986,
217
+ "step": 2400,
218
+ "token_acc": 0.9007656570069128
219
+ },
220
+ {
221
+ "epoch": 0.2671511006625347,
222
+ "grad_norm": 1.265625,
223
+ "learning_rate": 4.255917857747183e-06,
224
+ "loss": 0.3913582992553711,
225
+ "step": 2500,
226
+ "token_acc": 0.9015067434468572
227
+ },
228
+ {
229
+ "epoch": 0.27783714468903614,
230
+ "grad_norm": 1.46875,
231
+ "learning_rate": 4.193933428869592e-06,
232
+ "loss": 0.38879886627197263,
233
+ "step": 2600,
234
+ "token_acc": 0.9019667027692453
235
+ },
236
+ {
237
+ "epoch": 0.2885231887155375,
238
+ "grad_norm": 2.4375,
239
+ "learning_rate": 4.129961006610028e-06,
240
+ "loss": 0.3972347259521484,
241
+ "step": 2700,
242
+ "token_acc": 0.9001422292689183
243
+ },
244
+ {
245
+ "epoch": 0.2992092327420389,
246
+ "grad_norm": 1.2578125,
247
+ "learning_rate": 4.064075668743043e-06,
248
+ "loss": 0.3885000991821289,
249
+ "step": 2800,
250
+ "token_acc": 0.9027788505528588
251
+ },
252
+ {
253
+ "epoch": 0.3098952767685403,
254
+ "grad_norm": 1.265625,
255
+ "learning_rate": 3.996354738033166e-06,
256
+ "loss": 0.39520988464355467,
257
+ "step": 2900,
258
+ "token_acc": 0.9011279097672186
259
+ },
260
+ {
261
+ "epoch": 0.3205813207950417,
262
+ "grad_norm": 0.921875,
263
+ "learning_rate": 3.926877691489222e-06,
264
+ "loss": 0.3871217346191406,
265
+ "step": 3000,
266
+ "token_acc": 0.9022083313787118
267
+ },
268
+ {
269
+ "epoch": 0.33126736482154306,
270
+ "grad_norm": 1.375,
271
+ "learning_rate": 3.855726067090433e-06,
272
+ "loss": 0.3810308074951172,
273
+ "step": 3100,
274
+ "token_acc": 0.9034641205430405
275
+ },
276
+ {
277
+ "epoch": 0.3419534088480444,
278
+ "grad_norm": 1.5546875,
279
+ "learning_rate": 3.7829833680937804e-06,
280
+ "loss": 0.37891117095947263,
281
+ "step": 3200,
282
+ "token_acc": 0.9025715320536037
283
+ },
284
+ {
285
+ "epoch": 0.35263945287454584,
286
+ "grad_norm": 1.59375,
287
+ "learning_rate": 3.7087349650349234e-06,
288
+ "loss": 0.40724864959716794,
289
+ "step": 3300,
290
+ "token_acc": 0.8990014549892049
291
+ },
292
+ {
293
+ "epoch": 0.3633254969010472,
294
+ "grad_norm": 1.1171875,
295
+ "learning_rate": 3.6330679955376843e-06,
296
+ "loss": 0.39273796081542967,
297
+ "step": 3400,
298
+ "token_acc": 0.900933421493117
299
+ },
300
+ {
301
+ "epoch": 0.3740115409275486,
302
+ "grad_norm": 1.5,
303
+ "learning_rate": 3.556071262049684e-06,
304
+ "loss": 0.3975730514526367,
305
+ "step": 3500,
306
+ "token_acc": 0.8998019157822584
307
+ },
308
+ {
309
+ "epoch": 0.38469758495405,
310
+ "grad_norm": 2.140625,
311
+ "learning_rate": 3.477835127624158e-06,
312
+ "loss": 0.37645999908447264,
313
+ "step": 3600,
314
+ "token_acc": 0.9051004274755519
315
+ },
316
+ {
317
+ "epoch": 0.3953836289805514,
318
+ "grad_norm": 1.1796875,
319
+ "learning_rate": 3.39845140987024e-06,
320
+ "loss": 0.4185586166381836,
321
+ "step": 3700,
322
+ "token_acc": 0.8963374986700084
323
+ },
324
+ {
325
+ "epoch": 0.40606967300705277,
326
+ "grad_norm": 1.0390625,
327
+ "learning_rate": 3.318013273196187e-06,
328
+ "loss": 0.3819176483154297,
329
+ "step": 3800,
330
+ "token_acc": 0.9037970813391503
331
+ },
332
+ {
333
+ "epoch": 0.4167557170335542,
334
+ "grad_norm": 0.8828125,
335
+ "learning_rate": 3.2366151194720163e-06,
336
+ "loss": 0.4014278793334961,
337
+ "step": 3900,
338
+ "token_acc": 0.8989897804049096
339
+ },
340
+ {
341
+ "epoch": 0.42744176106005555,
342
+ "grad_norm": 1.2265625,
343
+ "learning_rate": 3.1543524772398464e-06,
344
+ "loss": 0.3903131103515625,
345
+ "step": 4000,
346
+ "token_acc": 0.9023779507523624
347
+ },
348
+ {
349
+ "epoch": 0.42744176106005555,
350
+ "eval_loss": 0.5559827089309692,
351
+ "eval_runtime": 1625.5851,
352
+ "eval_samples_per_second": 20.468,
353
+ "eval_steps_per_second": 1.28,
354
+ "eval_token_acc": 0.8759747254509199,
355
+ "step": 4000
356
+ },
357
+ {
358
+ "epoch": 0.43812780508655697,
359
+ "grad_norm": 1.5078125,
360
+ "learning_rate": 3.071321889601996e-06,
361
+ "loss": 0.3918019866943359,
362
+ "step": 4100,
363
+ "token_acc": 0.9021251892681234
364
+ },
365
+ {
366
+ "epoch": 0.4488138491130583,
367
+ "grad_norm": 1.3125,
368
+ "learning_rate": 2.9876208009183837e-06,
369
+ "loss": 0.39420204162597655,
370
+ "step": 4200,
371
+ "token_acc": 0.9012835119864208
372
+ },
373
+ {
374
+ "epoch": 0.45949989313955975,
375
+ "grad_norm": 3.140625,
376
+ "learning_rate": 2.9033474424462336e-06,
377
+ "loss": 0.39763076782226564,
378
+ "step": 4300,
379
+ "token_acc": 0.9007918217810081
380
+ },
381
+ {
382
+ "epoch": 0.4701859371660611,
383
+ "grad_norm": 1.1328125,
384
+ "learning_rate": 2.8186007170562607e-06,
385
+ "loss": 0.3739194869995117,
386
+ "step": 4400,
387
+ "token_acc": 0.9055805296059564
388
+ },
389
+ {
390
+ "epoch": 0.4808719811925625,
391
+ "grad_norm": 1.015625,
392
+ "learning_rate": 2.733480083160674e-06,
393
+ "loss": 0.3924733734130859,
394
+ "step": 4500,
395
+ "token_acc": 0.9005020191807561
396
+ },
397
+ {
398
+ "epoch": 0.4915580252190639,
399
+ "grad_norm": 1.140625,
400
+ "learning_rate": 2.648085437989179e-06,
401
+ "loss": 0.3818220520019531,
402
+ "step": 4600,
403
+ "token_acc": 0.9031686537753115
404
+ },
405
+ {
406
+ "epoch": 0.5022440692455653,
407
+ "grad_norm": 1.1875,
408
+ "learning_rate": 2.5625170003500053e-06,
409
+ "loss": 0.3947904968261719,
410
+ "step": 4700,
411
+ "token_acc": 0.9003777792274801
412
+ },
413
+ {
414
+ "epoch": 0.5129301132720667,
415
+ "grad_norm": 0.98046875,
416
+ "learning_rate": 2.4768751930135102e-06,
417
+ "loss": 0.39525905609130857,
418
+ "step": 4800,
419
+ "token_acc": 0.9012966596534361
420
+ },
421
+ {
422
+ "epoch": 0.523616157298568,
423
+ "grad_norm": 0.8984375,
424
+ "learning_rate": 2.391260524856434e-06,
425
+ "loss": 0.3915714645385742,
426
+ "step": 4900,
427
+ "token_acc": 0.9022641509433962
428
+ },
429
+ {
430
+ "epoch": 0.5343022013250694,
431
+ "grad_norm": 1.3671875,
432
+ "learning_rate": 2.3057734729050834e-06,
433
+ "loss": 0.3926304626464844,
434
+ "step": 5000,
435
+ "token_acc": 0.9013136721378624
436
+ },
437
+ {
438
+ "epoch": 0.5449882453515709,
439
+ "grad_norm": 1.34375,
440
+ "learning_rate": 2.2205143644159065e-06,
441
+ "loss": 0.3773085403442383,
442
+ "step": 5100,
443
+ "token_acc": 0.9033588372573927
444
+ },
445
+ {
446
+ "epoch": 0.5556742893780723,
447
+ "grad_norm": 1.109375,
448
+ "learning_rate": 2.135583259131831e-06,
449
+ "loss": 0.39842689514160157,
450
+ "step": 5200,
451
+ "token_acc": 0.9005042691639996
452
+ },
453
+ {
454
+ "epoch": 0.5663603334045736,
455
+ "grad_norm": 0.890625,
456
+ "learning_rate": 2.0510798318525585e-06,
457
+ "loss": 0.39046684265136716,
458
+ "step": 5300,
459
+ "token_acc": 0.9020825721592985
460
+ },
461
+ {
462
+ "epoch": 0.577046377431075,
463
+ "grad_norm": 0.92578125,
464
+ "learning_rate": 1.9671032554566205e-06,
465
+ "loss": 0.39618133544921874,
466
+ "step": 5400,
467
+ "token_acc": 0.9000712137530825
468
+ },
469
+ {
470
+ "epoch": 0.5877324214575764,
471
+ "grad_norm": 0.96484375,
472
+ "learning_rate": 1.8837520845124985e-06,
473
+ "loss": 0.3802578353881836,
474
+ "step": 5500,
475
+ "token_acc": 0.9039277912871451
476
+ },
477
+ {
478
+ "epoch": 0.5984184654840778,
479
+ "grad_norm": 2.265625,
480
+ "learning_rate": 1.8011241396153775e-06,
481
+ "loss": 0.40635391235351564,
482
+ "step": 5600,
483
+ "token_acc": 0.8981563749444691
484
+ },
485
+ {
486
+ "epoch": 0.6091045095105792,
487
+ "grad_norm": 1.53125,
488
+ "learning_rate": 1.7193163925852968e-06,
489
+ "loss": 0.36938583374023437,
490
+ "step": 5700,
491
+ "token_acc": 0.9048525545411809
492
+ },
493
+ {
494
+ "epoch": 0.6197905535370806,
495
+ "grad_norm": 1.0703125,
496
+ "learning_rate": 1.6384248526614122e-06,
497
+ "loss": 0.37309234619140624,
498
+ "step": 5800,
499
+ "token_acc": 0.9052111091372061
500
+ },
501
+ {
502
+ "epoch": 0.630476597563582,
503
+ "grad_norm": 1.515625,
504
+ "learning_rate": 1.558544453825951e-06,
505
+ "loss": 0.3821083068847656,
506
+ "step": 5900,
507
+ "token_acc": 0.9030327643665118
508
+ },
509
+ {
510
+ "epoch": 0.6411626415900834,
511
+ "grad_norm": 1.0390625,
512
+ "learning_rate": 1.479768943390068e-06,
513
+ "loss": 0.3839926910400391,
514
+ "step": 6000,
515
+ "token_acc": 0.9032827727270024
516
+ },
517
+ {
518
+ "epoch": 0.6411626415900834,
519
+ "eval_loss": 0.5562668442726135,
520
+ "eval_runtime": 1881.2991,
521
+ "eval_samples_per_second": 17.686,
522
+ "eval_steps_per_second": 1.106,
523
+ "eval_token_acc": 0.8759600467010075,
524
+ "step": 6000
525
+ },
526
+ {
527
+ "epoch": 0.6518486856165847,
528
+ "grad_norm": 1.5,
529
+ "learning_rate": 1.402190771972389e-06,
530
+ "loss": 0.3836984634399414,
531
+ "step": 6100,
532
+ "token_acc": 0.9032206080718131
533
+ },
534
+ {
535
+ "epoch": 0.6625347296430861,
536
+ "grad_norm": 1.703125,
537
+ "learning_rate": 1.3259009849993357e-06,
538
+ "loss": 0.381051025390625,
539
+ "step": 6200,
540
+ "token_acc": 0.9036385351259948
541
+ },
542
+ {
543
+ "epoch": 0.6732207736695875,
544
+ "grad_norm": 1.3125,
545
+ "learning_rate": 1.2509891158545895e-06,
546
+ "loss": 0.3822282028198242,
547
+ "step": 6300,
548
+ "token_acc": 0.9037642860928088
549
+ },
550
+ {
551
+ "epoch": 0.6839068176960889,
552
+ "grad_norm": 1.2265625,
553
+ "learning_rate": 1.1775430808030698e-06,
554
+ "loss": 0.3947774505615234,
555
+ "step": 6400,
556
+ "token_acc": 0.901950086240276
557
+ },
558
+ {
559
+ "epoch": 0.6945928617225903,
560
+ "grad_norm": 1.453125,
561
+ "learning_rate": 1.1056490758127697e-06,
562
+ "loss": 0.3984523391723633,
563
+ "step": 6500,
564
+ "token_acc": 0.9009766255095756
565
+ },
566
+ {
567
+ "epoch": 0.7052789057490917,
568
+ "grad_norm": 1.0625,
569
+ "learning_rate": 1.0353914753955118e-06,
570
+ "loss": 0.36463138580322263,
571
+ "step": 6600,
572
+ "token_acc": 0.9069917781813267
573
+ },
574
+ {
575
+ "epoch": 0.7159649497755931,
576
+ "grad_norm": 1.8984375,
577
+ "learning_rate": 9.668527335853722e-07,
578
+ "loss": 0.37465988159179686,
579
+ "step": 6700,
580
+ "token_acc": 0.9041866524545509
581
+ },
582
+ {
583
+ "epoch": 0.7266509938020944,
584
+ "grad_norm": 1.328125,
585
+ "learning_rate": 9.00113287170953e-07,
586
+ "loss": 0.3761553955078125,
587
+ "step": 6800,
588
+ "token_acc": 0.9049414331104472
589
+ },
590
+ {
591
+ "epoch": 0.7373370378285958,
592
+ "grad_norm": 1.2578125,
593
+ "learning_rate": 8.352514612951018e-07,
594
+ "loss": 0.3891106414794922,
595
+ "step": 6900,
596
+ "token_acc": 0.9018820430264891
597
+ },
598
+ {
599
+ "epoch": 0.7480230818550972,
600
+ "grad_norm": 0.92578125,
601
+ "learning_rate": 7.723433775328385e-07,
602
+ "loss": 0.3889284896850586,
603
+ "step": 7000,
604
+ "token_acc": 0.9016702188583
605
+ },
606
+ {
607
+ "epoch": 0.7587091258815987,
608
+ "grad_norm": 1.0703125,
609
+ "learning_rate": 7.114628645553845e-07,
610
+ "loss": 0.39446201324462893,
611
+ "step": 7100,
612
+ "token_acc": 0.9018489401695255
613
+ },
614
+ {
615
+ "epoch": 0.7693951699081,
616
+ "grad_norm": 1.0703125,
617
+ "learning_rate": 6.526813714851355e-07,
618
+ "loss": 0.38760986328125,
619
+ "step": 7200,
620
+ "token_acc": 0.902246221833607
621
+ },
622
+ {
623
+ "epoch": 0.7800812139346014,
624
+ "grad_norm": 1.0625,
625
+ "learning_rate": 5.960678840432629e-07,
626
+ "loss": 0.38959774017333987,
627
+ "step": 7300,
628
+ "token_acc": 0.9020091543079612
629
+ },
630
+ {
631
+ "epoch": 0.7907672579611028,
632
+ "grad_norm": 1.1640625,
633
+ "learning_rate": 5.416888435883516e-07,
634
+ "loss": 0.3816986083984375,
635
+ "step": 7400,
636
+ "token_acc": 0.9042744583449611
637
+ },
638
+ {
639
+ "epoch": 0.8014533019876042,
640
+ "grad_norm": 1.7265625,
641
+ "learning_rate": 4.896080691410895e-07,
642
+ "loss": 0.3822678756713867,
643
+ "step": 7500,
644
+ "token_acc": 0.9037521445862418
645
+ },
646
+ {
647
+ "epoch": 0.8121393460141055,
648
+ "grad_norm": 1.3125,
649
+ "learning_rate": 4.398866824865264e-07,
650
+ "loss": 0.3938971710205078,
651
+ "step": 7600,
652
+ "token_acc": 0.9006727321568306
653
+ },
654
+ {
655
+ "epoch": 0.822825390040607,
656
+ "grad_norm": 1.3359375,
657
+ "learning_rate": 3.9258303644179734e-07,
658
+ "loss": 0.3785970687866211,
659
+ "step": 7700,
660
+ "token_acc": 0.9045622587379151
661
+ },
662
+ {
663
+ "epoch": 0.8335114340671084,
664
+ "grad_norm": 1.421875,
665
+ "learning_rate": 3.4775264637349065e-07,
666
+ "loss": 0.3850682067871094,
667
+ "step": 7800,
668
+ "token_acc": 0.9020322018638262
669
+ },
670
+ {
671
+ "epoch": 0.8441974780936098,
672
+ "grad_norm": 1.0546875,
673
+ "learning_rate": 3.0544812504504114e-07,
674
+ "loss": 0.36783409118652344,
675
+ "step": 7900,
676
+ "token_acc": 0.9066081400907046
677
+ },
678
+ {
679
+ "epoch": 0.8548835221201111,
680
+ "grad_norm": 1.421875,
681
+ "learning_rate": 2.657191208706031e-07,
682
+ "loss": 0.39583698272705076,
683
+ "step": 8000,
684
+ "token_acc": 0.900744985000499
685
+ },
686
+ {
687
+ "epoch": 0.8548835221201111,
688
+ "eval_loss": 0.5561216473579407,
689
+ "eval_runtime": 1878.2163,
690
+ "eval_samples_per_second": 17.715,
691
+ "eval_steps_per_second": 1.107,
692
+ "eval_token_acc": 0.8760808641041333,
693
+ "step": 8000
694
+ }
695
+ ],
696
+ "logging_steps": 100,
697
+ "max_steps": 9358,
698
+ "num_input_tokens_seen": 0,
699
+ "num_train_epochs": 1,
700
+ "save_steps": 2000,
701
+ "stateful_callbacks": {
702
+ "TrainerControl": {
703
+ "args": {
704
+ "should_epoch_stop": false,
705
+ "should_evaluate": false,
706
+ "should_log": false,
707
+ "should_save": true,
708
+ "should_training_stop": false
709
+ },
710
+ "attributes": {}
711
+ }
712
+ },
713
+ "total_flos": 6.436080223138406e+18,
714
+ "train_batch_size": 16,
715
+ "trial_name": null,
716
+ "trial_params": null
717
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-8000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:859f9e6349e507974dc4ff76def0df20e49d08364f05bd0551845ae28b369820
3
+ size 7377
v0-20260123-014416/v1-20260123-200453/checkpoint-9358/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.11.1
v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [],
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": "^(encoder.*\\.(query|mlp.0|mlp.2|out|value|key)|adapter.*\\.(linear1|linear2))$",
23
+ "task_type": "CAUSAL_LM",
24
+ "use_dora": false,
25
+ "use_rslora": false
26
+ }
v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba70fd4ed51bb98671b7c6fd2d272db13f2f5c89bd3341778e92e85c67fb6bad
3
+ size 11988912
v0-20260123-014416/v1-20260123-200453/checkpoint-9358/additional_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"lora_dtype": null, "lorap_lr_ratio": null, "lorap_emb_lr": 1e-06}
v0-20260123-014416/v1-20260123-200453/checkpoint-9358/args.json ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": false,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 16,
10
+ "per_device_eval_batch_size": 16,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 4,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 5e-06,
18
+ "weight_decay": 0.01,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.95,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 1.0,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "cosine",
26
+ "lr_scheduler_kwargs": null,
27
+ "warmup_ratio": 0.02,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 100,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 2000.0,
39
+ "save_total_limit": 40,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": 42,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": true,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": -1,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": null,
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 2000.0,
65
+ "dataloader_num_workers": null,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF",
69
+ "disable_tqdm": null,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": "loss",
74
+ "greater_is_better": false,
75
+ "ignore_data_skip": false,
76
+ "fsdp": "",
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": null,
79
+ "fsdp_transformer_layer_cls_to_wrap": null,
80
+ "accelerator_config": {
81
+ "dispatch_batches": false
82
+ },
83
+ "deepspeed": null,
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "adamw_torch",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": null,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "dataloader_persistent_workers": false,
98
+ "skip_memory_metrics": true,
99
+ "use_legacy_prediction_loop": false,
100
+ "push_to_hub": false,
101
+ "resume_from_checkpoint": "/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000",
102
+ "hub_model_id": null,
103
+ "hub_strategy": "every_save",
104
+ "hub_token": null,
105
+ "hub_private_repo": null,
106
+ "hub_always_push": false,
107
+ "gradient_checkpointing": true,
108
+ "gradient_checkpointing_kwargs": "{\"use_reentrant\": false}",
109
+ "include_inputs_for_metrics": false,
110
+ "include_for_metrics": [],
111
+ "eval_do_concat_batches": true,
112
+ "fp16_backend": "auto",
113
+ "evaluation_strategy": "steps",
114
+ "push_to_hub_model_id": null,
115
+ "push_to_hub_organization": null,
116
+ "push_to_hub_token": null,
117
+ "mp_parameters": "",
118
+ "auto_find_batch_size": false,
119
+ "full_determinism": false,
120
+ "torchdynamo": null,
121
+ "ray_scope": "last",
122
+ "ddp_timeout": 18000000,
123
+ "torch_compile": false,
124
+ "torch_compile_backend": null,
125
+ "torch_compile_mode": null,
126
+ "dispatch_batches": null,
127
+ "split_batches": null,
128
+ "include_tokens_per_second": false,
129
+ "include_num_input_tokens_seen": false,
130
+ "neftune_noise_alpha": null,
131
+ "optim_target_modules": null,
132
+ "batch_eval_metrics": false,
133
+ "eval_on_start": false,
134
+ "use_liger_kernel": false,
135
+ "eval_use_gather_object": false,
136
+ "average_tokens_across_devices": false,
137
+ "sortish_sampler": false,
138
+ "predict_with_generate": false,
139
+ "generation_max_length": null,
140
+ "generation_num_beams": null,
141
+ "generation_config": null,
142
+ "tuner_backend": "peft",
143
+ "vit_gradient_checkpointing": null,
144
+ "router_aux_loss_coef": 0.0,
145
+ "enable_dft_loss": false,
146
+ "enable_channel_loss": false,
147
+ "check_model": true,
148
+ "acc_strategy": "token",
149
+ "train_dataloader_shuffle": true,
150
+ "max_epochs": null,
151
+ "aligner_lr": 1e-05,
152
+ "vit_lr": 5e-06,
153
+ "use_logits_to_keep": null,
154
+ "ds3_gather_for_generation": true,
155
+ "resume_only_model": false,
156
+ "optimizer": null,
157
+ "loss_type": null,
158
+ "metric": null,
159
+ "eval_use_evalscope": false,
160
+ "eval_dataset": [],
161
+ "eval_dataset_args": null,
162
+ "eval_limit": null,
163
+ "eval_generation_config": null,
164
+ "extra_eval_args": null,
165
+ "use_flash_ckpt": false,
166
+ "use_ray": false,
167
+ "ray_exp_name": null,
168
+ "device_groups": null,
169
+ "model": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
170
+ "model_type": "step_audio2_mini",
171
+ "model_revision": null,
172
+ "task_type": "causal_lm",
173
+ "torch_dtype": "bfloat16",
174
+ "attn_impl": null,
175
+ "new_special_tokens": [],
176
+ "num_labels": null,
177
+ "problem_type": null,
178
+ "rope_scaling": null,
179
+ "device_map": null,
180
+ "max_memory": {},
181
+ "max_model_len": null,
182
+ "local_repo_path": null,
183
+ "init_strategy": null,
184
+ "template": "step_audio2_mini",
185
+ "system": null,
186
+ "max_length": 16384,
187
+ "truncation_strategy": "delete",
188
+ "max_pixels": null,
189
+ "agent_template": null,
190
+ "norm_bbox": null,
191
+ "use_chat_template": true,
192
+ "padding_free": false,
193
+ "padding_side": "right",
194
+ "loss_scale": "default",
195
+ "sequence_parallel_size": 1,
196
+ "response_prefix": null,
197
+ "template_backend": "swift",
198
+ "dataset": [
199
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_train90.jsonl"
200
+ ],
201
+ "val_dataset": [
202
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
203
+ ],
204
+ "cached_dataset": [],
205
+ "cached_val_dataset": [],
206
+ "split_dataset_ratio": 0.0,
207
+ "dataset_num_proc": 1,
208
+ "load_from_cache_file": false,
209
+ "dataset_shuffle": true,
210
+ "val_dataset_shuffle": false,
211
+ "streaming": false,
212
+ "interleave_prob": null,
213
+ "stopping_strategy": "first_exhausted",
214
+ "shuffle_buffer_size": 1000,
215
+ "download_mode": "reuse_dataset_if_exists",
216
+ "columns": {},
217
+ "strict": false,
218
+ "model_name": null,
219
+ "model_author": null,
220
+ "custom_dataset_info": [],
221
+ "quant_method": null,
222
+ "quant_bits": null,
223
+ "hqq_axis": null,
224
+ "bnb_4bit_compute_dtype": "bfloat16",
225
+ "bnb_4bit_quant_type": "nf4",
226
+ "bnb_4bit_use_double_quant": true,
227
+ "bnb_4bit_quant_storage": null,
228
+ "max_new_tokens": 64,
229
+ "temperature": 0.0,
230
+ "top_k": null,
231
+ "top_p": null,
232
+ "repetition_penalty": null,
233
+ "num_beams": 1,
234
+ "stream": false,
235
+ "stop_words": [],
236
+ "logprobs": false,
237
+ "top_logprobs": null,
238
+ "ckpt_dir": null,
239
+ "lora_modules": [],
240
+ "train_type": "lora",
241
+ "adapters": [],
242
+ "external_plugins": [],
243
+ "model_kwargs": {},
244
+ "load_args": false,
245
+ "load_data_args": false,
246
+ "packing": false,
247
+ "packing_length": null,
248
+ "packing_num_proc": 1,
249
+ "lazy_tokenize": true,
250
+ "custom_register_path": [],
251
+ "use_hf": false,
252
+ "ignore_args_error": false,
253
+ "use_swift_lora": false,
254
+ "freeze_parameters": [],
255
+ "freeze_parameters_regex": null,
256
+ "freeze_parameters_ratio": 0.0,
257
+ "trainable_parameters": [],
258
+ "trainable_parameters_regex": null,
259
+ "freeze_llm": true,
260
+ "freeze_vit": false,
261
+ "freeze_aligner": false,
262
+ "target_modules": [
263
+ "all-linear"
264
+ ],
265
+ "target_regex": null,
266
+ "target_parameters": null,
267
+ "modules_to_save": [],
268
+ "lora_rank": 8,
269
+ "lora_alpha": 32,
270
+ "lora_dropout": 0.1,
271
+ "lora_bias": "none",
272
+ "lora_dtype": null,
273
+ "lorap_lr_ratio": null,
274
+ "use_rslora": false,
275
+ "use_dora": false,
276
+ "lora_ga_batch_size": 2,
277
+ "lora_ga_iters": 2,
278
+ "lora_ga_max_length": 1024,
279
+ "lora_ga_direction": "ArB2r",
280
+ "lora_ga_scale": "stable",
281
+ "lora_ga_stable_gamma": 16,
282
+ "init_weights": true,
283
+ "fourier_n_frequency": 2000,
284
+ "fourier_scaling": 300.0,
285
+ "boft_block_size": 4,
286
+ "boft_block_num": 0,
287
+ "boft_n_butterfly_factor": 1,
288
+ "boft_dropout": 0.0,
289
+ "vera_rank": 256,
290
+ "vera_projection_prng_key": 0,
291
+ "vera_dropout": 0.0,
292
+ "vera_d_initial": 0.1,
293
+ "adapter_act": "gelu",
294
+ "adapter_length": 128,
295
+ "use_galore": false,
296
+ "galore_target_modules": null,
297
+ "galore_rank": 128,
298
+ "galore_update_proj_gap": 50,
299
+ "galore_scale": 1.0,
300
+ "galore_proj_type": "std",
301
+ "galore_optim_per_parameter": false,
302
+ "galore_with_embedding": false,
303
+ "galore_quantization": false,
304
+ "galore_proj_quant": false,
305
+ "galore_proj_bits": 4,
306
+ "galore_proj_group_size": 256,
307
+ "galore_cos_threshold": 0.4,
308
+ "galore_gamma_proj": 2,
309
+ "galore_queue_size": 5,
310
+ "adalora_target_r": 8,
311
+ "adalora_init_r": 12,
312
+ "adalora_tinit": 0,
313
+ "adalora_tfinal": 0,
314
+ "adalora_deltaT": 1,
315
+ "adalora_beta1": 0.85,
316
+ "adalora_beta2": 0.85,
317
+ "adalora_orth_reg_weight": 0.5,
318
+ "llamapro_num_new_blocks": 4,
319
+ "llamapro_num_groups": null,
320
+ "lisa_activated_layers": 0,
321
+ "lisa_step_interval": 20,
322
+ "reft_layer_key": null,
323
+ "reft_layers": null,
324
+ "reft_rank": 4,
325
+ "reft_intervention_type": "LoreftIntervention",
326
+ "reft_args": null,
327
+ "swanlab_token": null,
328
+ "swanlab_project": null,
329
+ "swanlab_workspace": null,
330
+ "swanlab_exp_name": null,
331
+ "swanlab_lark_webhook_url": null,
332
+ "swanlab_lark_secret": null,
333
+ "swanlab_mode": "cloud",
334
+ "add_version": true,
335
+ "create_checkpoint_symlink": false,
336
+ "zero_hpz_partition_size": null,
337
+ "deepspeed_autotp_size": null,
338
+ "early_stop_interval": null,
339
+ "rank": -1,
340
+ "global_world_size": 1,
341
+ "local_world_size": 1,
342
+ "model_suffix": "Step-Audio-2-mini",
343
+ "model_info": "ModelInfo(model_type='step_audio2_mini', model_dir='/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini', torch_dtype=torch.bfloat16, max_model_len=16384, quant_method=None, quant_bits=None, rope_scaling=None, is_moe_model=False, is_multimodal=True, config=None, task_type='causal_lm', num_labels=None)",
344
+ "model_meta": "ModelMeta(model_type='step_audio2_mini', model_groups=[ModelGroup(models=[Model(ms_model_id='stepfun-ai/Step-Audio-2-mini', hf_model_id='stepfun-ai/Step-Audio-2-mini', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='step_audio2_mini', get_function=<function get_model_tokenizer_step_audio2_mini at 0x7fdf2b1c04c0>, model_arch=MultiModelKeys(arch_name='step_audio2_mini', 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', 'lm_head'], aligner=['adapter'], vision_tower=['encoder'], generator=[]), architectures=['StepAudio2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, is_reranker=False, task_type=None, ignore_patterns=None, requires=['transformers==4.53.3', 'torchaudio', 'librosa'], tags=['audio'])",
345
+ "model_dir": "/data/pangkaiyu/dg/Step-Audio2/Step-Audio-2-mini",
346
+ "_val_dataset_exists": [
347
+ "/data/pangkaiyu/dg/make_jsonl_py_file/lora_merged_0122_55+5+5+1_signal_val5.jsonl"
348
+ ],
349
+ "hub": "<class 'swift.hub.hub.MSHub'>",
350
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.STEPS: 'steps'>, prediction_loss_only=False, per_device_train_batch_size=16, per_device_eval_batch_size=16, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=4, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=5e-06, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=1.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.02, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/v1-20260123-200453/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=100, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.STEPS: 'steps'>, save_steps=2000, save_total_limit=40, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend=None, tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=2000, dataloader_num_workers=1, dataloader_prefetch_factor=10, past_index=-1, run_name='step-audio2-mini-enc+align-whole0122_signal_new_dpdc_lora-1gpu-bs16*4_1_gckF', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, fsdp_transformer_layer_cls_to_wrap=None, 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), deepspeed=None, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint='/data/pangkaiyu/dg/output_step_audio2_mini-encoder+align-whole0122_signal_new1_dpdc-lora-1gpu-bs16*4_1_gckF/v0-20260123-014416/checkpoint-4000', hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs={'use_reentrant': False}, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', evaluation_strategy='steps', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=18000000, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, dispatch_batches=None, split_batches=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=True, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, aligner_lr=1e-05, vit_lr=5e-06, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer='multimodal', loss_type=None, metric=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, use_flash_ckpt=False, sft_alpha=0, chord_sft_dataset=[], chord_sft_per_device_train_batch_size=None, chord_enable_phi_function=False, chord_mu_warmup_steps=None, chord_mu_decay_steps=None, chord_mu_peak=None, chord_mu_valley=None, train_type='lora', local_repo_path=None, galore_config=None, padding_side='right', padding_free=False, task_type='causal_lm', problem_type=None)"
351
+ }