From 0c7c75a75433dd9b8b4fb833ad7039e1408519aa Mon Sep 17 00:00:00 2001
From: Nick White <git@njw.name>
Date: Tue, 9 Apr 2019 10:44:11 +0100
Subject: Modify traintessv4.sh to include step to construct final training

---
 traintessv4.sh | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'traintessv4.sh')

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"
-- 
cgit v1.2.1-24-ge1ad