Update preprocess.sh

JDACS4C-IMPROVE · Dec 20, 2023 · 04f7d37 · 04f7d37
1 parent 2bac956
commit 04f7d37
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/templates/preprocess.sh b/src/templates/preprocess.sh
@@ -31,15 +31,21 @@ fi
 
 if [ $# -eq 2 ] ; then
         CANDLE_DATA_DIR=$1 ; shift
-        CONFIG_FILE=$1 ; shift
-        CMD="python ${CANDLE_MODEL} --config_file ${CONFIG_FILE}"
+
+	# if $2 is a file, then set candle_config
+	if [ -f $CANDLE_DATA_DIR/$1 ] ; then
+ 		CONFIG_FILE=$1 ; shift
+        	CMD="python ${CANDLE_MODEL} --config_file ${CONFIG_FILE}"
+	else
+ 		CMD="python ${CANDLE_MODEL} $@"
+
         echo "CMD = $CMD"
 
 elif [ $# -ge 3 ] ; then
-        CUDA_VISIBLE_DEVICES=$1 ; shift
+
         CANDLE_DATA_DIR=$1 ; shift
 
-        # if $3 is a file, then set candle_config
+        # if $2 is a file, then set candle_config
         if [ -f $CANDLE_DATA_DIR/$1 ] ; then
 		echo "$1 is a file"
                 CANDLE_CONFIG=$1 ; shift