| {"pretrained_model": "openai/whisper-small.en", "dataset_path": "./data", "output_dir": "./fine-tuned-whisper", "num_train_epochs": 1, "train_batch_size": 64, "gradient_accumulation_steps": 1, "eval_batch_size": 8, "max_learning_rate": 1e-05, "warmup_steps": 0, "max_steps": 0, "eval_steps": null, "gradient_checkpointing": true, "fp16": true, "evaluation_strategy": "epoch", "logging_steps": 500, "save_steps": 1000, "dataset": "mixed", "real": 10.0, "fake": 50.0, "patience": 5} |