From 1eb2cecd0ad4f15f09d183cee8a3c3fbefbbfa9b Mon Sep 17 00:00:00 2001
From: Daniel Ecer <de-code@users.noreply.github.com>
Date: Tue, 1 Aug 2017 11:51:47 +0100
Subject: [PATCH] added dataset suffix and eval set size configuration
---
eval.sh | 2 +-
prepare-shell.sh | 9 +++++----
2 files changed, 6 insertions(+), 5 deletions(-)
diff --git a/eval.sh b/eval.sh
index 4696a0f..3c05333 100755
--- a/eval.sh
+++ b/eval.sh
@@ -15,7 +15,7 @@ COMMON_ARGS=(
--color_map "${CONFIG_PATH}/${COLOR_MAP_FILENAME}"
--use_separate_channels $USE_SEPARATE_CHANNELS
--batch_size 10
- --eval_set_size 10
+ --eval_set_size $EVAL_SET_SIZE
--max_steps 0
--write_predictions
${TRAINING_ARGS[@]}
diff --git a/prepare-shell.sh b/prepare-shell.sh
index aecad55..38fcf1c 100755
--- a/prepare-shell.sh
+++ b/prepare-shell.sh
@@ -13,6 +13,7 @@ export LOCAL_PATH_ROOT="./.models"
export BUCKET="gs://${PROJECT}-ml"
export COLOR_MAP_FILENAME="color_map.conf"
export USE_SEPARATE_CHANNELS=true
+export DATASET_SUFFIX=
export USE_CLOUD=false
@@ -40,16 +41,16 @@ fi
# cloud paths
export GCS_SUB_PROJECT_PATH="${BUCKET}/${SUB_PROJECT_NAME}"
export GCS_PATH="${GCS_SUB_PROJECT_PATH}/${MODEL_NAME}/${VERSION_NAME}"
-export GCS_DATA_PATH="${GCS_PATH}/data"
+export GCS_DATA_PATH="${GCS_PATH}/data${DATASET_SUFFIX}"
export GCS_CONFIG_PATH="${GCS_PATH}/config"
-export GCS_PREPROC_PATH="${GCS_PATH}/preproc"
+export GCS_PREPROC_PATH="${GCS_PATH}/preproc${DATASET_SUFFIX}"
export GCS_TRAIN_MODEL_PATH="${GCS_PATH}${TRAINING_SUFFIX}/training"
# local paths
export LOCAL_MODEL_PATH="${LOCAL_PATH_ROOT}/${MODEL_NAME}/${VERSION_NAME}"
-export LOCAL_DATA_PATH="${LOCAL_MODEL_PATH}/data"
+export LOCAL_DATA_PATH="${LOCAL_MODEL_PATH}/data${DATASET_SUFFIX}"
export LOCAL_CONFIG_PATH="."
-export LOCAL_PREPROC_PATH="${LOCAL_MODEL_PATH}/preproc"
+export LOCAL_PREPROC_PATH="${LOCAL_MODEL_PATH}/preproc${DATASET_SUFFIX}"
export LOCAL_TRAIN_MODEL_PATH="${LOCAL_MODEL_PATH}${TRAINING_SUFFIX}/training"
echo "USE_CLOUD: $USE_CLOUD"
--
GitLab