From b0e91a5994699217cfc44f44d5564dc5e100cb0e Mon Sep 17 00:00:00 2001 From: Cyrille Nofficial Date: Thu, 6 Oct 2022 14:05:49 +0200 Subject: [PATCH] feat(training): bash script to run training --- run_training.sh | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/run_training.sh b/run_training.sh index 67b1a41..fbe236f 100755 --- a/run_training.sh +++ b/run_training.sh @@ -3,8 +3,8 @@ set +e set +x -RECORDS_PATH=~/robocar/record-sim4-2 -#TRAINING_OPTS="--horizon=20" +RECORDS_PATH=~/src/robocars/data/viva20/viva12/ +#TRAINING_OPTS="--horizon=50" TRAINING_OPTS="" MODEL_TYPE="categorical" #MODEL_TYPE="linear" @@ -12,8 +12,9 @@ IMG_WIDTH=160 IMG_HEIGHT=120 HORIZON=20 -TRAINING_DATA_DIR=/tmp/data -TRAINING_OUTPUT_DIR=/tmp/output +TRAINING_DIR=~/src/robocars/trainings +TRAINING_DATA_DIR=${TRAINING_DIR}/data +TRAINING_OUTPUT_DIR=${TRAINING_DIR}/output TRAIN_ARCHIVE=${TRAINING_DATA_DIR}/train.zip ####################### @@ -30,10 +31,10 @@ go run ./cmd/rc-tools training archive \ -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 \ +podman run --rm -it \ + -v /trainings/data:/opt/ml/input/data/train \ + -v /trainings/output:/opt/ml/model/ \ + localhost/tensorflow_without_gpu:old \ python /opt/ml/code/train.py \ --img_height=${IMG_HEIGHT} \ --img_width=${IMG_WIDTH} \