Upload full run folder: v0-20260123-013837
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- v0-20260123-014416/args.json +351 -0
- v0-20260123-014416/checkpoint-2000/README.md +202 -0
- v0-20260123-014416/checkpoint-2000/adapter_config.json +26 -0
- v0-20260123-014416/checkpoint-2000/adapter_model.safetensors +3 -0
- v0-20260123-014416/checkpoint-2000/additional_config.json +1 -0
- v0-20260123-014416/checkpoint-2000/args.json +351 -0
- v0-20260123-014416/checkpoint-2000/optimizer.pt +3 -0
- v0-20260123-014416/checkpoint-2000/rng_state.pth +3 -0
- v0-20260123-014416/checkpoint-2000/scheduler.pt +3 -0
- v0-20260123-014416/checkpoint-2000/trainer_state.json +210 -0
- v0-20260123-014416/checkpoint-2000/training_args.bin +3 -0
- v0-20260123-014416/checkpoint-4000/README.md +202 -0
- v0-20260123-014416/checkpoint-4000/adapter_config.json +26 -0
- v0-20260123-014416/checkpoint-4000/adapter_model.safetensors +3 -0
- v0-20260123-014416/checkpoint-4000/additional_config.json +1 -0
- v0-20260123-014416/checkpoint-4000/args.json +351 -0
- v0-20260123-014416/checkpoint-4000/optimizer.pt +3 -0
- v0-20260123-014416/checkpoint-4000/rng_state.pth +3 -0
- v0-20260123-014416/checkpoint-4000/scheduler.pt +3 -0
- v0-20260123-014416/checkpoint-4000/trainer_state.json +379 -0
- v0-20260123-014416/checkpoint-4000/training_args.bin +3 -0
- v0-20260123-014416/logging.jsonl +44 -0
- v0-20260123-014416/v0-20260123-194351/args.json +351 -0
- v0-20260123-014416/v0-20260123-194351/logging.jsonl +1 -0
- v0-20260123-014416/v1-20260123-200453/args.json +351 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/README.md +202 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_config.json +26 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/adapter_model.safetensors +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/additional_config.json +1 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/args.json +351 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/optimizer.pt +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/rng_state.pth +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/scheduler.pt +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/trainer_state.json +548 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-6000/training_args.bin +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/README.md +202 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_config.json +26 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/adapter_model.safetensors +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/additional_config.json +1 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/args.json +351 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/optimizer.pt +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/rng_state.pth +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/scheduler.pt +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/trainer_state.json +717 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-8000/training_args.bin +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-9358/README.md +202 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_config.json +26 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-9358/adapter_model.safetensors +3 -0
- v0-20260123-014416/v1-20260123-200453/checkpoint-9358/additional_config.json +1 -0
- 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 |
+
}
|