mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-10 02:22:17 +00:00
61 lines
1.7 KiB
Bash
Executable File
61 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -ex
|
|
|
|
cd /icefall
|
|
export PYTHONPATH=/icefall:$PYTHONPATH
|
|
python3 -c "import torch; print(torch.__file__)"
|
|
python3 -c "import torchaudio; print(torchaudio.__version__)"
|
|
python3 -c "import icefall; print(icefall.__file__)"
|
|
|
|
cd egs/librispeech/ASR
|
|
|
|
# We don't download the LM file since it is so large that it will
|
|
# cause OOM error for CI later.
|
|
mkdir -p download/lm
|
|
pushd download/lm
|
|
wget -q http://www.openslr.org/resources/11/librispeech-vocab.txt
|
|
wget -q http://www.openslr.org/resources/11/librispeech-lexicon.txt
|
|
wget -q http://www.openslr.org/resources/11/librispeech-lm-norm.txt.gz
|
|
ls -lh
|
|
gunzip librispeech-lm-norm.txt.gz
|
|
|
|
ls -lh
|
|
popd
|
|
|
|
pushd download/
|
|
wget -q https://huggingface.co/csukuangfj/librispeech-for-ci/resolve/main/LibriSpeech.tar.bz2
|
|
tar xf LibriSpeech.tar.bz2
|
|
rm LibriSpeech.tar.bz2
|
|
|
|
cd LibriSpeech
|
|
ln -s train-clean-100 train-clean-360
|
|
ln -s train-other-500 train-other-500
|
|
popd
|
|
|
|
mkdir -p data/manifests
|
|
|
|
lhotse prepare librispeech -j 2 -p dev-clean -p dev-other -p test-clean -p test-other -p train-clean-100 download/LibriSpeech data/manifests
|
|
ls -lh data/manifests
|
|
|
|
./local/compute_fbank_librispeech.py --dataset "dev-clean dev-other test-clean test-other train-clean-100" --perturb-speed False
|
|
ls -lh data/fbank
|
|
|
|
./prepare.sh --stage 5 --stop-stage 6
|
|
|
|
./zipformer/train.py \
|
|
--world-size 1 \
|
|
--num-epochs 1 \
|
|
--start-epoch 1 \
|
|
--use-fp16 0 \
|
|
--exp-dir zipformer/exp-small \
|
|
--causal 0 \
|
|
--num-encoder-layers 1,1,1,1,1,1 \
|
|
--feedforward-dim 64,96,96,96,96,96 \
|
|
--encoder-dim 32,64,64,64,64,64 \
|
|
--encoder-unmasked-dim 32,32,32,32,32,32 \
|
|
--base-lr 0.04 \
|
|
--full-libri 0 \
|
|
--enable-musan 0 \
|
|
--max-duration 30 \
|
|
--print-diagnostics 1
|