update run.sh
This commit is contained in:
parent
325ada3037
commit
1c213aceac
@ -6,7 +6,7 @@
|
|||||||
|
|
||||||
export CUDA_VISIBLE_DEVICES="0"
|
export CUDA_VISIBLE_DEVICES="0"
|
||||||
|
|
||||||
stage=2
|
stage=-1
|
||||||
stop_stage=4
|
stop_stage=4
|
||||||
num_keywords=11
|
num_keywords=11
|
||||||
|
|
||||||
@ -18,7 +18,7 @@ gpu_id=4
|
|||||||
checkpoint=
|
checkpoint=
|
||||||
dir=exp/mdtc_debug
|
dir=exp/mdtc_debug
|
||||||
|
|
||||||
num_average=1
|
num_average=10
|
||||||
score_checkpoint=$dir/avg_${num_average}.pt
|
score_checkpoint=$dir/avg_${num_average}.pt
|
||||||
|
|
||||||
# your data dir
|
# your data dir
|
||||||
@ -88,7 +88,7 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
|
|||||||
--num ${num_average} \
|
--num ${num_average} \
|
||||||
--val_best
|
--val_best
|
||||||
|
|
||||||
# Compute posterior score
|
# Testing
|
||||||
result_dir=$dir/test_$(basename $score_checkpoint)
|
result_dir=$dir/test_$(basename $score_checkpoint)
|
||||||
mkdir -p $result_dir
|
mkdir -p $result_dir
|
||||||
python kws/bin/test.py --gpu 3 \
|
python kws/bin/test.py --gpu 3 \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user