File size: 441 Bytes
f6dd949 |
1 |
{"pretrained_model": "openai/whisper-medium.en", "dataset_path": "./data", "output_dir": "./fine-tuned-whisper", "num_train_epochs": 0, "train_batch_size": 32, "gradient_accumulation_steps": 1, "eval_batch_size": 16, "max_learning_rate": 1e-05, "warmup_steps": 500, "max_steps": 10000, "eval_steps": 500, "gradient_checkpointing": true, "fp16": true, "evaluation_strategy": "steps", "logging_steps": 100, "save_steps": 500, "dataset": "all"} |