From 80ab5f30a8a3b7542aebfab7e3e344ba4fcb5bbf Mon Sep 17 00:00:00 2001
From: Henry Addison <henryaddison@users.noreply.github.com>
Date: Wed, 18 Sep 2024 13:21:32 +0100
Subject: [PATCH] update tensorboard helper so it can pick up lots of dirs
 based on input

and add a version for JASMIN
---
 bin/bp/{bp-tb => tb} |  4 ++--
 bin/jasmin/tb        | 12 ++++++++++++
 2 files changed, 14 insertions(+), 2 deletions(-)
 rename bin/bp/{bp-tb => tb} (83%)
 create mode 100755 bin/jasmin/tb

diff --git a/bin/bp/bp-tb b/bin/bp/tb
similarity index 83%
rename from bin/bp/bp-tb
rename to bin/bp/tb
index 1e7fcbec..53116178 100755
--- a/bin/bp/bp-tb
+++ b/bin/bp/tb
@@ -3,10 +3,10 @@
 
 set -euo pipefail
 
-run_name=$1
+runs_dir=$1
 host=${2:-bp}
 
 port=$(shuf -i 6000-9999 -n 1)
 echo "Using port ${port} on ${host}"
 
-ssh -t -L ${port}:localhost:${port} ${host} "~/mambaforge/envs/cuda-downscaling/bin/tensorboard --port=${port} --logdir=/user/work/vf20964/workdirs/${run_name}/tensorboard"
+ssh -t -L ${port}:localhost:${port} ${host} "~/mambaforge/envs/cuda-downscaling/bin/tensorboard --port=${port} --logdir=/user/work/vf20964/workdirs/${runs_dir}"
diff --git a/bin/jasmin/tb b/bin/jasmin/tb
new file mode 100755
index 00000000..37b0fe9c
--- /dev/null
+++ b/bin/jasmin/tb
@@ -0,0 +1,12 @@
+#!/usr/bin/bash
+# run tensorboard on a login node on a random port
+
+set -euo pipefail
+
+runs_dir=$1
+host=${2:-sci2.jasmin}
+
+port=$(shuf -i 6000-9999 -n 1)
+echo "Using port ${port} on ${host}"
+
+ssh -t -L ${port}:localhost:${port} ${host} "~/mambaforge/envs/mv-mlde/bin/tensorboard --port=${port} --logdir=/gws/nopw/j04/bris_climdyn/henrya/jasmin-active/workdirs/${runs_dir}"