From eda5f69f2d902d7f5d357519e7f20aae38f656b2 Mon Sep 17 00:00:00 2001
From: Daniel <daniel@mail.de>
Date: Thu, 18 Jun 2020 15:20:28 +0200
Subject: [PATCH] Install checkpoint converting tool.

---
 Dockerfile.train.tmpl | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Dockerfile.train.tmpl b/Dockerfile.train.tmpl
index 426a80f8..c9a92117 100644
--- a/Dockerfile.train.tmpl
+++ b/Dockerfile.train.tmpl
@@ -46,4 +46,8 @@ RUN pip3 install --upgrade pip==20.0.2 wheel==0.34.2 setuptools==46.1.3
 #    we don't want to break that
 RUN DS_NODECODER=y DS_NOTENSORFLOW=y pip3 install --upgrade --force-reinstall -e .
 
+# Tool to convert output graph for inference
+RUN python3 /DeepSpeech/util/taskcluster.py --source tensorflow --branch r1.15 \
+        --artifact convert_graphdef_memmapped_format  --target /DeepSpeech/
+
 RUN ./bin/run-ldc93s1.sh