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 K=18
FINAL_K=12 FINAL_K=12
STOP=1E-3 STOP=1E-1
echo "*********" echo "*********"
echo "Direct" echo "Direct"
@ -53,5 +53,5 @@ else
echo "final stage $FINAL_K elements" echo "final stage $FINAL_K elements"
t=$(mktemp) t=$(mktemp)
extract -e `expr $FINAL_K - 1` -t $K $VQ_NAME'_s3.f32' $t 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 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/ PATH=$PATH:/home/david/codec2-dev/build_linux/misc/
VQTRAIN=$PATH/vqtrain
EXTRACT=$PATH/extract 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 "*********"
echo "Pred 1" echo "Pred 1"
echo "*********" echo "*********"
$EXTRACT all_speech_features.f32 all_speech_pred1.f32 0 17 10 0.9 echo "weighting dctLy[0] ...."
$VQTRAIN all_speech_pred1.f32 18 2048 /dev/null s11.f32 t=$(mktemp)
$VQTRAIN s11.f32 18 2048 /dev/null s12.f32 extract all_speech_features.f32 $t 0 17 10 1.0 1
$VQTRAIN s12.f32 18 2048 /dev/null s13.f32 cat $t | ./weight > $VQ_NAME'_s0.f32'
#$VQTRAIN s13.f32 18 512 /dev/null s14.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 if [ $# -lt 1 ]; then
echo "usage: ./train_pred2.sh [-w] VQprefix" echo "usage: ./train_pred2.sh [-w] VQprefix"
echo " $ ./train_pred2.sh pred2_v1" echo " $ ./train_pred2.sh pred2_v1"
echo " -w weight dctLy[0]"
exit 1 exit 1
fi fi
for i in "$@"
do
case $i in
-w)
WEIGHT=1
echo "WEIGHT!"
shift # past argument=value
;;
esac
done
VQ_NAME=$1 VQ_NAME=$1
echo $VQ_NAME echo $VQ_NAME
K=18 K=18
STOP=1E-3 STOP=1E-2
echo "*********" echo "*********"
echo "Pred 2" echo "Pred 2"
echo "*********" echo "*********"
if [ -z "$WEIGHT" ]; then echo "weighting dctLy[0] ...."
extract all_speech_features_5e6.f32 $VQ_NAME'_s0.f32' 0 17 10 0.9 2 t=$(mktemp)
else extract all_speech_features_5e6.f32 $t 0 17 10 1 2
echo "weighting dctLy[0] ...." cat $t | ./weight > $VQ_NAME'_s0.f32'
t=$(mktemp)
extract all_speech_features_5e6.f32 $t 0 17 10 0.9 2
cat $t | ./weight > $VQ_NAME'_s0.f32'
fi
vqtrain $VQ_NAME'_s0.f32' $K 2048 $VQ_NAME'_stage1.f32' -r $VQ_NAME'_s1.f32' -s $STOP 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'_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'_s2.f32' $K 2048 $VQ_NAME'_stage3.f32' -r $VQ_NAME'_s3.f32' -s $STOP