sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

slurm-pretrain: prepare

Browse source
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
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
aimodel/slurm-pretrain.job
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;