train_conf.json

[
    {
        "per_device_train_batch_size": 64,
        "gradient_accumulation_steps": 1,
        "per_device_eval_batch_size": 32,
        "evaluation_strategy": "steps",
        "max_steps": 2000,
        "save_steps": 100,
        "eval_steps": 100,
        "logging_steps": 50,
        "learning_rate": 0.0001,
        "weight_decay": 0,
        "warmup_steps": 600,
        "save_total_limit": 1,
        "metric_for_best_model": "wer",
        "greater_is_better": false
    },
    {
        "model_path": "facebook/wav2vec2-large-lv60",
        "problem_type": "single_label_classification",
        "task_type": "asr",
        "model_type": "baseline",
        "final_dropout": 0.0,
        "ctc_zero_infinity": true,
        "layerdrop": 0.1,
        "activation_dropout": 0.1,
        "mask_time_prob": 0.065,
        "mask_time_length": 10,
        "mask_feature_prob": 0.015,
        "mask_feature_length": 64
    }
]