merge queue training and sampling jobs on JASMIN into one

henryaddison · Aug 9, 2024 · 3badd1c · 3badd1c
1 parent 7a846ad
commit 3badd1c
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 16 deletions.
diff --git a/bin/jasmin/queue-mlde b/bin/jasmin/queue-mlde
@@ -0,0 +1,30 @@
+#!/bin/bash
+# Script for queueing a model job on LOTUS on JASMIN via lotus-wrapper script
+
+set -euo pipefail
+
+SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+
+smem=128G
+stime=1-00:00:00
+
+while getopts ":m:t:" opt; do
+  case ${opt} in
+    m)
+      smem=${OPTARG}
+      ;;
+    t)
+      stime=${OPTARG}
+      ;;
+    \? )
+      # echo "Invalid option: -${OPTARG}" 1>&2
+     ;;
+    : )
+      echo "Invalid option: $OPTARG requires an argument" 1>&2
+      exit 1
+      ;;
+  esac
+done
+shift "$((OPTIND -1))"
+
+sbatch --parsable --gres=gpu:1 --partition=orchid --account=orchid --time=${stime} --mem=${smem} -- ${SCRIPT_DIR}/lotus-wrapper $@
diff --git a/bin/jasmin/queue-sampling b/bin/jasmin/queue-sampling
diff --git a/bin/jasmin/queue-training b/bin/jasmin/queue-training