Skip to content

Commit

Permalink
use checkpoints for real
Browse files Browse the repository at this point in the history
  • Loading branch information
wasertech committed Apr 26, 2022
1 parent 1cb6aa0 commit d7ba737
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions bin/run-ci-lm-opt.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,16 +16,15 @@ LM_NUM_TRIALS=1
scorer_filepath="./data/smoke_test/pruned_lm.scorer"
test_filepath="./data/smoke_test/ldc93s1_flac.csv"

checkpoint_dirpath="./data"
mkdir -p checkpoint_dirpath
checkpoint_dir="${HOME}/.local/share/stt/ldc93s1"

# Force only one visible device because we have a single-sample dataset
# and when trying to run on multiple devices (like GPUs), this will break

export CUDA_VISIBLE_DEVICES=0

python -m coqui_stt_training.util.lm_optimize \
--scorer_path $scorer_filepath \
--checkpoint_dir "$checkpoint_dirpath" \
--checkpoint_dir "$checkpoint_dir" \
--test_files $test_filepath \
--n_trials $LM_NUM_TRIALS \
--lm_alpha_max ${LM_ALPHA_MAX} \
Expand Down

0 comments on commit d7ba737

Please sign in to comment.