N-Shot Benchmarking of Whisper
- Install dependencies in your environment:
pip install -r requirementx.txt
- Run train.py as:
python train.py --training_args
OR - Run train.sh as:
bash train.sh
OR - Submit a slurm job for training:
sbatch --job_args train.sh --training_args
- You can use multiple launcher including
deepspeed
,torchrun
, andpython
. - Remove
--deepspeed "ds_config2.json"
and change launcher fromdeepspeed
totorchrun
orpython
if you don't want to use deepspeed.
@article{talafha2023n,
title={N-Shot Benchmarking of Whisper on Diverse Arabic Speech Recognition},
author={Talafha, Bashar and Waheed, Abdul and Abdul-Mageed, Muhammad},
journal={arXiv preprint arXiv:2306.02902},
year={2023}
}