mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
15 lines
518 B
Bash
Executable File
15 lines
518 B
Bash
Executable File
|
|
export PYTHONPATH=$PYTHONPATH:/workspace/asr/icefall
|
|
# pip install k2==1.24.3.dev20230524+cuda11.8.torch2.0.1 -f https://k2-fsa.github.io/k2/cuda.html
|
|
# pip install -r whisper/requirements.txt
|
|
|
|
method=mask_predict
|
|
# method=cif_ar_distill_embedding
|
|
torchrun --nproc_per_node 8 ./parawhisper/train.py \
|
|
--max-duration 200 \
|
|
--exp-dir parawhisper/exp_large_v2_${method} \
|
|
--model-name large-v2 \
|
|
--manifest-dir data/fbank \
|
|
--method $method \
|
|
--deepspeed \
|
|
--deepspeed_config ./whisper/ds_config_zero1.json |