sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

slurm-pretrain-plot: add ARGS

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-27 18:41:35 +01:00
parent f4d1d1d77e
commit 2cd59a01a5
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/slurm-pretrain-predict.job
View file

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