mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 06:34:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

add fairseq2 require

Browse Source
This commit is contained in:
Yuekai Zhang 2023-09-07 01:21:43 -07:00
parent 3a7ad277ad
commit cbc3852876
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/aishell/ASR/run.sh
View File

@ -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

3
egs/aishell/ASR/seamlessm4t/requirements.txt
View File

@ -1,5 +1,6 @@
k2
#k2
kaldialign
lhotse
sentencepiece
tensorboard
fairseq2