diff --git a/egs/aishell/ASR/run_whisper.sh b/egs/aishell/ASR/run_whisper.sh index d96089db1..f97e44af2 100644 --- a/egs/aishell/ASR/run_whisper.sh +++ b/egs/aishell/ASR/run_whisper.sh @@ -1,7 +1,9 @@ -pip install -r whisper/requirements.txt pip install k2==1.24.3.dev20230524+cuda11.8.torch2.0.1 -f https://k2-fsa.github.io/k2/cuda.html -export PYTHONPATH=$PYTHONPATH:/mnt/samsung-t7/yuekai/asr/icefall +pip install -r whisper/requirements.txt +export PYTHONPATH=$PYTHONPATH:/workspace/icefall +#export PYTHONPATH=$PYTHONPATH:/lustre/fsw/sa/yuekaiz/asr/icefall +#export PYTHONPATH=$PYTHONPATH:/mnt/samsung-t7/yuekai/asr/icefall torchrun --nproc-per-node 8 whisper/train.py --use-fp16 1 --max-duration 20 --base-lr 1e-5 --exp-dir whisper/exp_medimum --start-epoch 1 diff --git a/egs/aishell/ASR/whisper/requirements.txt b/egs/aishell/ASR/whisper/requirements.txt index e0c221ded..6edea0afc 100644 --- a/egs/aishell/ASR/whisper/requirements.txt +++ b/egs/aishell/ASR/whisper/requirements.txt @@ -1,6 +1,7 @@ k2 kaldialign -lhotse +lhotse==1.18 +#git+https://github.com/lhotse-speech/lhotse sentencepiece tensorboard librosa