{
  "model_path": "./models/MOSS-TTS-Nano",
  "codec_path": "./models/MOSS-Audio-Tokenizer-Nano",
  "train_jsonl": "train_with_codes_v2.jsonl",
  "output_dir": "./checkpoints/iaratts-sft-v2",
  "max_length": 1024,
  "per_device_batch_size": 8,
  "gradient_accumulation_steps": 4,
  "learning_rate": 3e-05,
  "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": "cosine",
  "num_epochs": 5,
  "max_train_steps": null,
  "max_grad_norm": 1.0,
  "logging_steps": 20,
  "save_every_epochs": 1,
  "num_workers": 2,
  "mixed_precision": "bf16",
  "attn_implementation": "sdpa",
  "channelwise_loss_weight": "1,32",
  "seed": 42,
  "resolved_warmup_steps": 39,
  "resolved_channelwise_loss_weight": [
    1.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0,
    2.0
  ],
  "global_batch_size": 32,
  "records_paths": [
    "/root/iaratts/MOSS-TTS-Nano/train_with_codes_v2.jsonl"
  ],
  "saved_global_step": 765,
  "saved_epoch": 5,
  "saved_at": "2026-04-27 06:43:06",
  "checkpoint_dir": "checkpoints/iaratts-sft-v2/checkpoint-last"
}