From 2cd59a01a517c89f924a90d14a476846b063cb49 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 27 Sep 2022 18:41:35 +0100 Subject: [PATCH] slurm-pretrain-plot: add ARGS --- aimodel/slurm-pretrain-predict.job | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/slurm-pretrain-predict.job b/aimodel/slurm-pretrain-predict.job index 4fe26b1..3a2c0f1 100755 --- a/aimodel/slurm-pretrain-predict.job +++ b/aimodel/slurm-pretrain-predict.job @@ -24,6 +24,7 @@ show_help() { echo -e " INPUT The path to the input file (.jsonl.gz) containing the embedded data to plot (see the pretrain-predict subcommand for embedding data)" >&2; echo -e " CHECKPOINT The filepath to the checkpoint (.hdf5) file to load" >&2; echo -e " POSTFIX Arbitrary string to add to filename." >&2; + echo -e " ARGS Any additional args to pass to pretrain-plot." >&2; echo -e "" >&2; echo -e "The code used to identify the run is taken automatically from the filename of the config file." >&2; exit; @@ -69,7 +70,7 @@ echo ">>> Installing requirements"; conda run -n py38 pip install -r requirements.txt; echo ">>> Training model"; #shellcheck disable=SC2086 -conda run -n py38 /usr/bin/env time -v src/index.py pretrain-predict --input "${INPUT}" --output "${filepath_output}" -c "${CHECKPOINT}" +conda run -n py38 /usr/bin/env time -v src/index.py pretrain-predict --input "${INPUT}" --output "${filepath_output}" -c "${CHECKPOINT}" ${ARGS} echo ">>> exited with code $?"; INPUT="${filepath_output}" POSTFIX="${POSTFIX}" sbatch slurm-pretrain-plot.job