From a3f03b6d8dab2a45876c898e3342a3dca77bf9f5 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 2 Sep 2022 19:05:18 +0100 Subject: [PATCH] slurm-pretrain: prepare --- aimodel/slurm-pretrain.job | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/aimodel/slurm-pretrain.job b/aimodel/slurm-pretrain.job index 6380206..bcbcc76 100755 --- a/aimodel/slurm-pretrain.job +++ b/aimodel/slurm-pretrain.job @@ -29,10 +29,10 @@ show_help() { exit; } -CONFIG="${CONFIG:-configs/clip.toml}"; +INPUT="${INPUT:-$HOME/rainfallwater_records_tfrecord}"; -if [[ -z "${CONFIG}" ]]; then - echo -e "Error: No CONFIG environment variable specified.\n" >&2; +if [[ -z "${INPUT}" ]]; then + echo -e "Error: No INPUT environment variable specified.\n" >&2; show_help; exit 0; fi @@ -44,8 +44,7 @@ if [[ ! -d "${INPUT}" ]]; then fi -CODE="$(basename "${CONFIG}")"; -CODE="${CODE%.*}"; +CODE="$(date +"%Y-%m-%d")_pretrain_contrast"; if [[ -n "${POSTFIX}" ]]; then echo -e ">>> Applying postfix of ${POSTFIX}" >&2;