File size: 1,233 Bytes
40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 da065c3 40147b8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | {
"model_path": "./models/MOSS-TTS-Nano",
"codec_path": "./models/MOSS-Audio-Tokenizer-Nano",
"train_jsonl": "/workspace/train_ref_with_codes.jsonl",
"output_dir": "/workspace/output/moss_tts_ref_v6",
"max_length": 1024,
"per_device_batch_size": 4,
"gradient_accumulation_steps": 4,
"learning_rate": 0.0001,
"weight_decay": 0.1,
"adam_beta1": 0.9,
"adam_beta2": 0.95,
"adam_eps": 1e-08,
"warmup_steps": 0,
"warmup_ratio": 0.05,
"lr_scheduler_type": "linear",
"num_epochs": 50,
"max_train_steps": null,
"max_grad_norm": 1.0,
"logging_steps": 10,
"save_every_epochs": 1,
"num_workers": 0,
"mixed_precision": "fp16",
"attn_implementation": "sdpa",
"channelwise_loss_weight": "1,8",
"seed": 42,
"resolved_warmup_steps": 6523,
"resolved_channelwise_loss_weight": [
1.0,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5,
0.5
],
"global_batch_size": 16,
"records_paths": [
"/workspace/train_ref_with_codes.jsonl"
],
"saved_global_step": 75661,
"saved_epoch": 29,
"saved_at": "2026-05-08 06:33:19",
"checkpoint_dir": "/workspace/output/moss_tts_ref_v6/checkpoint-epoch-29"
} |