Modify traintessv4.sh to include step to construct final training

1 files changed, 8 insertions, 0 deletions

diff --git a/traintessv4.sh b/traintessv4.sh
index 9e4e8f4..92da4be 100755
--- a/traintessv4.sh
+++ b/traintessv4.sh
@@ -79,6 +79,7 @@ combine_lang_model --input_unicharset "$name/unicharset" --script_dir "$name/sta
 
 mkdir -p "$name/checkpoint"
 
+echo "Starting training"
 lstmtraining \
 	--traineddata "$name/starter/$name/$name.traineddata" \
 	--old_traineddata "$name/orig/orig.traineddata" \
@@ -90,3 +91,10 @@ lstmtraining \
 	--eval_listfile "$name/eval/list" \
 	--max_iterations $iterations \
 	$extra || exit 1
+
+echo "Saving training"
+lstmtraining \
+	--stop_training \
+	--continue_from "$name/checkpoint/${name}_checkpoint" \
+	--traineddata "$name/starter/$name/$name.traineddata" \
+	--model_output "$name/$name.traineddata"