slurm-pretrain: prepare

This commit is contained in:
Starbeamrainbowlabs 2022-09-02 19:05:18 +01:00
parent 1d1533d160
commit a3f03b6d8d
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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;