feat(training): bash script to run training
This commit is contained in:
parent
c4ff4b46b0
commit
b0e91a5994
@ -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} \
|
||||
|
Loading…
Reference in New Issue
Block a user