From 58048b90d7caacfd4feabe4ff05a9ea6b4c79205 Mon Sep 17 00:00:00 2001 From: Roman Grundkiewicz Date: Wed, 23 Feb 2022 04:18:03 -0800 Subject: Allow setting models/ or data/ directories --- run_mrt.sh | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/run_mrt.sh b/run_mrt.sh index bdf2dee..3c3da4e 100755 --- a/run_mrt.sh +++ b/run_mrt.sh @@ -14,6 +14,8 @@ # - NUM_DEVICES - maximum number of GPU devices to be used # - TIMEOUT - maximum duration for execution of a single test in the format # accepted by the timeout command; set to 0 to disable +# - MODELS - path to the directory with models, default: ./models +# - DATA - path to the directory with data, default: ./data SHELL=/bin/bash @@ -32,8 +34,13 @@ log "Running on $(hostname) as process $$" export MRT_ROOT="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" export MRT_TOOLS=$MRT_ROOT/tools export MRT_MARIAN="$( realpath ${MARIAN:-$MRT_ROOT/../build} )" -export MRT_MODELS=$MRT_ROOT/models -export MRT_DATA=$MRT_ROOT/data + +# Print folders which contain models and data for regression tests +export MRT_MODELS="$( realpath ${MODELS:-$MRT_ROOT/models} )" +export MRT_DATA="$( realpath ${DATA:-$MRT_ROOT/data} )" + +log "Using models from: $MRT_MODELS" +log "Using data from: $MRT_DATA" # Try adding build/ to MARIAN for backward compatibility if [[ ! -e $MRT_MARIAN/marian-decoder ]]; then -- cgit v1.2.3