slurm pretrain predict: add $OUTPUT

This commit is contained in:
Starbeamrainbowlabs 2022-09-27 18:58:02 +01:00
parent 7d0e3913ae
commit 2d24174e0a
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -22,6 +22,7 @@ show_help() {
echo -e "" >&2; echo -e "" >&2;
echo -e "....where:" >&2; echo -e "....where:" >&2;
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 " 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 " OUTPUT Optional. The filepath/dir to write output to." >&2;
echo -e " CHECKPOINT The filepath to the checkpoint (.hdf5) file to load" >&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 " POSTFIX Arbitrary string to add to filename." >&2;
echo -e " ARGS Any additional args to pass to pretrain-plot." >&2; echo -e " ARGS Any additional args to pass to pretrain-plot." >&2;
@ -62,7 +63,11 @@ echo -e ">>> Input dirpath: ${INPUT}" >&2;
echo -e ">>> Code: ${CODE}" >&2; echo -e ">>> Code: ${CODE}" >&2;
echo -e ">>> Additional args: ${ARGS}"; echo -e ">>> Additional args: ${ARGS}";
filepath_output="output/rainfallwater_records_embed_$(date -u --rfc-3339=date)${CODE}.jsonl.gz"; if [[ -n "${OUTPUT}" ]]; then
filepath_output="${OUTPUT}";
else
filepath_output="output/rainfallwater_records_embed_$(date -u --rfc-3339=date)${CODE}.jsonl.gz";
fi
export PATH=$HOME/software/bin:$PATH; export PATH=$HOME/software/bin:$PATH;