diff --git a/egs/aishell/ASR/run.sh b/egs/aishell/ASR/run.sh index 8c0770b05..7d40ea887 100644 --- a/egs/aishell/ASR/run.sh +++ b/egs/aishell/ASR/run.sh @@ -1,4 +1,8 @@ -export CUDA_VISIBLE_DEVICES="2,3" -export PYTHONPATH=$PYTHONPATH:/mnt/samsung-t7/yuekai/asr/icefall -torchrun --nproc-per-node 2 seamlessm4t/train2.py --use-fp16 1 --max-duration 20 +#export CUDA_VISIBLE_DEVICES="2,3" +pip install -r seamlessm4t/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:/lustre/fsw/sa/yuekaiz/asr/icefall +export PYTHONPATH=$PYTHONPATH:/lustre/fsw/sa/yuekaiz/asr/seamless_communication/src +export TORCH_HOME=/lustre/fsw/sa/yuekaiz/asr/hub +torchrun --nproc-per-node 8 seamlessm4t/train2.py --use-fp16 1 --max-duration 300 diff --git a/egs/aishell/ASR/seamlessm4t/requirements.txt b/egs/aishell/ASR/seamlessm4t/requirements.txt index 16b1725ee..7647735da 100644 --- a/egs/aishell/ASR/seamlessm4t/requirements.txt +++ b/egs/aishell/ASR/seamlessm4t/requirements.txt @@ -1,5 +1,6 @@ -k2 +#k2 kaldialign lhotse sentencepiece tensorboard +fairseq2