{
    "training": {
        "seed": 42,
        "logging_steps": 10,
        "eval_steps": 50,
        "gradient_accumulation_steps": 2,
        "gradient_clip_value": 1.0,
        "learning_rate": 3e-04,
        "betas": [0.9, 0.95],
        "warmup_ratio": 0.02,
        "batch_size": 4,
        "weight_decay": 0.1,
        "precision": "bf16",
        "strategy": "ddp",
        "gradient_checkpointing": true,
        "num_workers": 2,
        "max_training_steps": 500,
        "min_lr_ratio": 0.1
    },
    "modeling": {
        "parameters": {
            "codebook_size": 65536,
            "max_seq_len": 2048,
            "model_name": "meta-llama/Llama-3.2-1B-Instruct",
            "enable_text_normalization": false
        }
    },
    "checkpointing": {
        "save_steps": 100,
        "collect_health_stats": true,
        "save_intermediate_generations": false,
        "only_load_model_weights": false,
        "keep_only_last_n_checkpoints": 3
    },
    "train_weighted_datasets": {
        "/path/to/your/pt_audio_dataset": 1.0,
        "/path/to/your/text_corpus[text]": 0.1
    },
    "val_weighted_datasets": {
        "/path/to/your/pt_audio_dataset": 1.0
    },
    "dataset": {
        "allowed_languages": [],
        "min_dnsmos_score": 0,
        "min_sample_rate": 0,
        "enable_rlhf_training": false
    }
}
