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} \