-
Start docker 启动命令可参考: README.md
-
Prepare model
apt install git-lfs
git lfs install
git clone https://huggingface.co/openai/whisper-small
# git clone https://huggingface.co/openai/whisper-large
# git clone https://huggingface.co/openai/whisper-tiny
- Download dataset
python download.py- Prerequisites
pip install -r requirements.txt- Train
# Single GPU
bash train_whisper.sh
# Multi-GPU DDP
bash dist_train_whisper.sh- Test
python test_whisper.py --models_dir <models-dir> --data_dir <datasets-dir>