{
    "mon": {
        "dataset_path": "data/baby_cry_detection",
        "output_dir": "distilhubert-finetuned-cry-detector",
        "training_args": {
            "num_train_epochs": 6, 
            "learning_rate": 0.0001,
            "warmup_ratio": 0.001,
            "output_dir": "distilhubert-finetuned-cry-detector",
            "eval_strategy": "epoch",
            "save_strategy": "epoch",
            "lr_scheduler_type": "cosine",
            "auto_find_batch_size": true,
            "per_device_train_batch_size": 8,
            "per_device_eval_batch_size": 8,
            "gradient_accumulation_steps": 8,
            "gradient_checkpointing": true,
            "load_best_model_at_end": true,
            "greater_is_better": true,
            "metric_for_best_model": "accuracy",
            "optim": "adamw_torch",
            "hub_strategy": "checkpoint",
            "report_to": "tensorboard",
            "full_determinism": true,
            "seed": 123,
            "data_seed":123
        }
    },
    "class": {
        "dataset_path": "data/mixed_data",
        "output_dir": "distilhubert-finetuned-mixed-data",
        "training_args": {
            "num_train_epochs": 10,
            "learning_rate": 0.0001,
            "warmup_ratio": 0.001,
            "output_dir": "distilhubert-finetuned-mixed-data",
            "eval_strategy": "epoch",
            "save_strategy": "epoch",
            "lr_scheduler_type": "cosine",
            "auto_find_batch_size": true,
            "per_device_train_batch_size": 8,
            "per_device_eval_batch_size": 8,
            "gradient_accumulation_steps": 8,
            "gradient_checkpointing": true,
            "load_best_model_at_end": true,
            "greater_is_better": true,
            "metric_for_best_model": "accuracy",
            "optim": "adamw_torch",
            "hub_strategy": "checkpoint",
            "report_to": "tensorboard",
            "full_determinism": true,
            "seed": 123,
            "data_seed":123
        }
    }
}