Merge pull request #47 from lissyx/package

Package TensorFlow and TFLite models and checkpoints

DeepSpeech/Dockerfile.train.fr

@@ -69,7 +69,8 @@ RUN apt-get -qq update && apt-get -qq install -y --no-install-recommends \
     pixz \
     sox \
     libsox-fmt-all \
-    locales locales-all
+    locales locales-all \
+    zip
 
 RUN groupadd -g 999 trainer && \
     adduser --system --uid 999 --group trainer

DeepSpeech/package.sh

@@ -0,0 +1,30 @@
+#!/bin/bash
+
+set -xe
+
+pushd /mnt
+	zip -r9 --junk-paths             \
+		model_tensorflow_fr.zip  \
+		models/output_graph.pbmm \
+		models/alphabet.txt      \
+		lm/lm.binary             \
+		lm/trie
+
+	zip -r9 --junk-paths               \
+		model_tflite_fr.zip        \
+		models/output_graph.tflite \
+		models/alphabet.txt        \
+		lm/lm.binary               \
+		lm/trie
+
+	all_checkpoint_path=""
+	for ckpt in $(grep 'all_model_checkpoint_paths' checkpoints/checkpoint | cut -d'"' -f2);
+	do
+		all_checkpoint_path="${all_checkpoint_path} ${ckpt}.*"
+	done;
+
+	zip -r9 --junk-paths           \
+		checkpoint_fr.zip      \
+		checkpoints/checkpoint \
+		${all_checkpoint_path}
+popd

DeepSpeech/run.sh

@@ -22,3 +22,5 @@ generate_alphabet.sh
 build_lm.sh
 
 train_fr.sh
+
+package.sh

DeepSpeech/train_fr.sh

@@ -9,7 +9,9 @@ pushd $HOME/ds/
 
 	mkdir -p /mnt/sources/feature_cache || true
 
-	if [ -f "/transfer-checkpoint/checkpoint" ]; then
+        # Do not overwrite checkpoint file if model already exist: we will likely
+	# only package
+	if [ -f "/transfer-checkpoint/checkpoint" -a ! -f "/mnt/models/output_graph.pb" ]; then
 		echo "Using checkpoint from ${TRANSFER_CHECKPOINT}"
 		cp -a /transfer-checkpoint/* /mnt/checkpoints/
 	fi;