experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Exercise training graph inference/Flashlight decoder in extra training tests

Browse Source
This commit is contained in:
Reuben Morais 2021-10-30 14:59:22 +02:00
parent 755fb81a62
commit 04f62ac9f7
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
bin/run-ci-ldc93s1_singleshotinference.sh
View File

@ -14,7 +14,8 @@ fi;
# and when trying to run on multiple devices (like GPUs), this will break # and when trying to run on multiple devices (like GPUs), this will break
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python -u train.py --alphabet_config_path "data/alphabet.txt" \ python -m coqui_stt_training.train \
--alphabet_config_path "data/alphabet.txt" \
--show_progressbar false --early_stop false \ --show_progressbar false --early_stop false \
--train_files ${ldc93s1_csv} --train_batch_size 1 \ --train_files ${ldc93s1_csv} --train_batch_size 1 \
--dev_files ${ldc93s1_csv} --dev_batch_size 1 \ --dev_files ${ldc93s1_csv} --dev_batch_size 1 \
@ -24,8 +25,15 @@ python -u train.py --alphabet_config_path "data/alphabet.txt" \
--learning_rate 0.001 --dropout_rate 0.05 \ --learning_rate 0.001 --dropout_rate 0.05 \
--scorer_path 'data/smoke_test/pruned_lm.scorer' --scorer_path 'data/smoke_test/pruned_lm.scorer'
python -u train.py --alphabet_config_path "data/alphabet.txt" \ python -m coqui_stt_training.training_graph_inference \
--n_hidden 100 \ --n_hidden 100 \
--checkpoint_dir '/tmp/ckpt' \ --checkpoint_dir '/tmp/ckpt' \
--scorer_path 'data/smoke_test/pruned_lm.scorer' \ --scorer_path 'data/smoke_test/pruned_lm.scorer' \
--one_shot_infer 'data/smoke_test/LDC93S1.wav' --one_shot_infer 'data/smoke_test/LDC93S1.wav'
python -m coqui_stt_training.training_graph_inference_flashlight \
--n_hidden 100 \
--checkpoint_dir '/tmp/ckpt' \
--scorer_path 'data/smoke_test/pruned_lm.scorer' \
--vocab_file 'data/smoke_test/vocab.pruned.txt' \
--one_shot_infer 'data/smoke_test/LDC93S1.wav'

3
ci_scripts/train-extra-tests.sh
View File

@ -69,3 +69,6 @@ time ./bin/run-ci-ldc93s1_checkpoint_bytes.sh
# Training with args set via initialize_globals_from_args() # Training with args set via initialize_globals_from_args()
time python ./bin/run-ldc93s1.py time python ./bin/run-ldc93s1.py
# Training graph inference
time ./bin/run-ci-ldc93s1_singleshotinference.sh