From e13d992b0d84e1e8c5a894aa35fb61bab9edfc42 Mon Sep 17 00:00:00 2001
From: David <david@rowetel.com>
Date: Thu, 21 Mar 2019 20:14:38 +1030
Subject: [PATCH] latest training scripts

---
 train_direct.sh |  4 ++--
 train_pred1.sh  | 35 ++++++++++++++++++++++++++---------
 train_pred2.sh  | 25 +++++--------------------
 3 files changed, 33 insertions(+), 31 deletions(-)

diff --git a/train_direct.sh b/train_direct.sh
index f48a761..879c273 100755
--- a/train_direct.sh
+++ b/train_direct.sh
@@ -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
diff --git a/train_pred1.sh b/train_pred1.sh
index 9742d9d..3694252 100755
--- a/train_pred1.sh
+++ b/train_pred1.sh
@@ -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 
+
diff --git a/train_pred2.sh b/train_pred2.sh
index fe5b9e7..298620b 100755
--- a/train_pred2.sh
+++ b/train_pred2.sh
@@ -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