robocars/robocar-tools
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat(training): bash script to run training

Browse Source
This commit is contained in:
Cyrille Nofficial 2022-10-06 14:05:49 +02:00
parent c4ff4b46b0
commit b0e91a5994
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
run_training.sh
View File

@ -3,8 +3,8 @@
set +e set +e
set +x set +x
RECORDS_PATH=~/robocar/record-sim4-2 RECORDS_PATH=~/src/robocars/data/viva20/viva12/
#TRAINING_OPTS="--horizon=20" #TRAINING_OPTS="--horizon=50"
TRAINING_OPTS="" TRAINING_OPTS=""
MODEL_TYPE="categorical" MODEL_TYPE="categorical"
#MODEL_TYPE="linear" #MODEL_TYPE="linear"
@ -12,8 +12,9 @@ IMG_WIDTH=160
IMG_HEIGHT=120 IMG_HEIGHT=120
HORIZON=20 HORIZON=20
TRAINING_DATA_DIR=/tmp/data TRAINING_DIR=~/src/robocars/trainings
TRAINING_OUTPUT_DIR=/tmp/output TRAINING_DATA_DIR=${TRAINING_DIR}/data
TRAINING_OUTPUT_DIR=${TRAINING_DIR}/output
TRAIN_ARCHIVE=${TRAINING_DATA_DIR}/train.zip TRAIN_ARCHIVE=${TRAINING_DATA_DIR}/train.zip
####################### #######################
@ -30,10 +31,10 @@ go run ./cmd/rc-tools training archive \
-image-width ${IMG_WIDTH} -image-width ${IMG_WIDTH}
printf "\n\nRun training\n\n" printf "\n\nRun training\n\n"
podman run --rm -it \ podman run --rm -it \
-v /tmp/data:/opt/ml/input/data/train \ -v /trainings/data:/opt/ml/input/data/train \
-v /tmp/output:/opt/ml/model/ \ -v /trainings/output:/opt/ml/model/ \
localhost/tensorflow_without_gpu \ localhost/tensorflow_without_gpu:old \
python /opt/ml/code/train.py \ python /opt/ml/code/train.py \
--img_height=${IMG_HEIGHT} \ --img_height=${IMG_HEIGHT} \
--img_width=${IMG_WIDTH} \ --img_width=${IMG_WIDTH} \