latest training scripts

pull/1/head
David 2019-03-21 20:14:38 +10:30
parent 446b5afd13
commit e13d992b0d
3 changed files with 33 additions and 31 deletions

View File

@ -27,7 +27,7 @@ echo $VQ_NAME
K=18
FINAL_K=12
STOP=1E-3
STOP=1E-1
echo "*********"
echo "Direct"
@ -53,5 +53,5 @@ else
echo "final stage $FINAL_K elements"
t=$(mktemp)
extract -e `expr $FINAL_K - 1` -t $K $VQ_NAME'_s3.f32' $t
vqtrain $t $FINAL_K 2048 $VQ_NAME'_stage4.f32' -r $VQ_NAME'_s5.f32' -s $STOP
vqtrain $t $FINAL_K 2048 $VQ_NAME'_stage4.f32' -r $VQ_NAME'_s5.f32' -s $STOP -t $K
fi

View File

@ -1,14 +1,31 @@
#!/bin/sh
#!/bin/sh -x
# train_pred2.sh
# David Rowe Jan 2019
# Train multi-stage VQ for LPCNet
PATH=/home/david/codec2-dev/build_linux/misc/
VQTRAIN=$PATH/vqtrain
EXTRACT=$PATH/extract
PATH=$PATH:/home/david/codec2-dev/build_linux/misc/
if [ $# -lt 1 ]; then
echo "usage: ./train_pred1.sh [-w] VQprefix"
echo " $ ./train_pred1.sh pred1_v1"
exit 1
fi
VQ_NAME=$1
echo $VQ_NAME
K=18
STOP=1E-2
echo "*********"
echo "Pred 1"
echo "*********"
$EXTRACT all_speech_features.f32 all_speech_pred1.f32 0 17 10 0.9
$VQTRAIN all_speech_pred1.f32 18 2048 /dev/null s11.f32
$VQTRAIN s11.f32 18 2048 /dev/null s12.f32
$VQTRAIN s12.f32 18 2048 /dev/null s13.f32
#$VQTRAIN s13.f32 18 512 /dev/null s14.f32
echo "weighting dctLy[0] ...."
t=$(mktemp)
extract all_speech_features.f32 $t 0 17 10 1.0 1
cat $t | ./weight > $VQ_NAME'_s0.f32'
vqtrain $VQ_NAME'_s0.f32' $K 2048 $VQ_NAME'_stage1.f32' -r $VQ_NAME'_s1.f32' -s $STOP
vqtrain $VQ_NAME'_s1.f32' $K 2048 $VQ_NAME'_stage2.f32' -r $VQ_NAME'_s2.f32' -s $STOP
vqtrain $VQ_NAME'_s2.f32' $K 2048 $VQ_NAME'_stage3.f32' -r $VQ_NAME'_s3.f32' -s $STOP
vqtrain $VQ_NAME'_s3.f32' $K 2048 $VQ_NAME'_stage4.f32' -r $VQ_NAME'_s4.f32' -s $STOP

View File

@ -8,37 +8,22 @@ PATH=$PATH:/home/david/codec2-dev/build_linux/misc/
if [ $# -lt 1 ]; then
echo "usage: ./train_pred2.sh [-w] VQprefix"
echo " $ ./train_pred2.sh pred2_v1"
echo " -w weight dctLy[0]"
exit 1
fi
for i in "$@"
do
case $i in
-w)
WEIGHT=1
echo "WEIGHT!"
shift # past argument=value
;;
esac
done
VQ_NAME=$1
echo $VQ_NAME
K=18
STOP=1E-3
STOP=1E-2
echo "*********"
echo "Pred 2"
echo "*********"
if [ -z "$WEIGHT" ]; then
extract all_speech_features_5e6.f32 $VQ_NAME'_s0.f32' 0 17 10 0.9 2
else
echo "weighting dctLy[0] ...."
t=$(mktemp)
extract all_speech_features_5e6.f32 $t 0 17 10 0.9 2
cat $t | ./weight > $VQ_NAME'_s0.f32'
fi
echo "weighting dctLy[0] ...."
t=$(mktemp)
extract all_speech_features_5e6.f32 $t 0 17 10 1 2
cat $t | ./weight > $VQ_NAME'_s0.f32'
vqtrain $VQ_NAME'_s0.f32' $K 2048 $VQ_NAME'_stage1.f32' -r $VQ_NAME'_s1.f32' -s $STOP
vqtrain $VQ_NAME'_s1.f32' $K 2048 $VQ_NAME'_stage2.f32' -r $VQ_NAME'_s2.f32' -s $STOP
vqtrain $VQ_NAME'_s2.f32' $K 2048 $VQ_NAME'_stage3.f32' -r $VQ_NAME'_s3.f32' -s $STOP