| { | |
| "pocket_tts_language": "portuguese", | |
| "erinome_pt": "/root/encoded_1500.pt", | |
| "output_dir": "/workspace/checkpoints", | |
| "epochs": 4, | |
| "micro_batch_size": 2, | |
| "grad_accum": 16, | |
| "learning_rate": 5e-05, | |
| "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": true, | |
| "rewind_threshold": 5.0, | |
| "curriculum": "", | |
| "multi_dataset_weights": null | |
| } |