v3_config.json

{
  "pocket_tts_language": "portuguese",
  "erinome_pt": "/root/encoded_1500.pt",
  "output_dir": "/workspace/lr_low_ckpt",
  "epochs": 1,
  "micro_batch_size": 2,
  "grad_accum": 16,
  "learning_rate": 5e-07,
  "weight_decay": 0.01,
  "warmup_steps": 200,
  "bf16": true,
  "seed": 42,
  "save_every_steps": 25,
  "log_every_steps": 25,
  "freeze_backbone_layers": 4,
  "flow_loss_weight": 1.0,
  "eos_loss_weight": 0.1,
  "grad_clip": 1.0,
  "auto_lr_rewind": false,
  "rewind_threshold": 5.0,
  "curriculum": "",
  "multi_dataset_weights": null
}