diff --git a/aimodel/slurm-pretrain-predict.job b/aimodel/slurm-pretrain-predict.job
index 023ce04..4b17197 100755
--- a/aimodel/slurm-pretrain-predict.job
+++ b/aimodel/slurm-pretrain-predict.job
@@ -22,6 +22,7 @@ show_help() {
 	echo -e "" >&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 "    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 "    POSTFIX     Arbitrary string to add to filename." >&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 ">>> 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;