From 988abb715feb5d67dca1d7d82ca834babc622921 Mon Sep 17 00:00:00 2001
From: Cyrille Nofficial <cynoffic@cyrilix.fr>
Date: Fri, 10 Jun 2022 12:35:01 +0200
Subject: [PATCH] add script to run training

---
 run_training.sh | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100755 run_training.sh

diff --git a/run_training.sh b/run_training.sh
new file mode 100755
index 0000000..3abf1e6
--- /dev/null
+++ b/run_training.sh
@@ -0,0 +1,47 @@
+#! /bin/bash
+
+set +e
+set +x
+
+RECORDS_PATH=~/robocar/record-sim4-2
+#TRAINING_OPTS="--horizon=20"
+TRAINING_OPTS=""
+#MODEL_TYPE="categorical"
+MODEL_TYPE="linear"
+IMG_WIDTH=160
+IMG_HEIGHT=120
+HORIZON=20
+
+TRAINING_DATA_DIR=/tmp/data
+TRAINING_OUTPUT_DIR=/tmp/output
+TRAIN_ARCHIVE=${TRAINING_DATA_DIR}/train.zip
+
+#######################
+
+rm -rf ${TRAINING_DATA_DIR} ${TRAINING_OUTPUT_DIR}
+mkdir -p ${TRAINING_DATA_DIR}
+mkdir -p ${TRAINING_OUTPUT_DIR}
+
+printf "Build archive %s\n\n" "${TRAIN_ARCHIVE}"
+go run ./cmd/rc-tools training archive \
+            -record-path ${RECORDS_PATH} \
+            -output ${TRAIN_ARCHIVE} \
+            -image-height ${IMG_HEIGHT} \
+            -image-width ${IMG_WIDTH}
+
+printf "\n\nRun training\n\n"
+podman run --rm -it \
+            -v /tmp/data:/opt/ml/input/data/train \
+            -v /tmp/output:/opt/ml/model/ \
+            localhost/tensorflow_without_gpu \
+                python /opt/ml/code/train.py \
+                      --img_height=${IMG_HEIGHT} \
+                      --img_width=${IMG_WIDTH} \
+                      --batch_size=32 \
+                      --model_type=${MODEL_TYPE} \
+                      --horizon=${HORIZON} \
+                      ${TRAINING_OPTS}
+
+printf "\n\nConvert model\n\n"
+edgetpu_compiler -o ${TRAINING_OUTPUT_DIR} ${TRAINING_OUTPUT_DIR}/model_*.tflite
+