From a78b8789c0376806d291c2e87ceca12c85d90425 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 2 Jul 2021 15:59:05 +0200
Subject: [PATCH] Training script with slurm scheduler.

---
 Scripts/train_slurm.sh | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 Scripts/train_slurm.sh

diff --git a/Scripts/train_slurm.sh b/Scripts/train_slurm.sh
new file mode 100644
index 0000000..5eeb0ea
--- /dev/null
+++ b/Scripts/train_slurm.sh
@@ -0,0 +1,30 @@
+#!/bin/bash
+
+# Options for SLURM sbatch
+
+#SBATCH --job-name=train-climax
+
+#SBATCH --output=train-climax-%j.out
+
+#SBATCH --error=train-climax-%j.err
+
+#SBATCH --partition gpu
+
+#SBATCH --nodes 1
+
+#SBATCH --mem=64G
+
+#SBATCH --gres=gpu:1
+
+#SBATCH --ntasks-per-node=1
+
+source /home/clusterusers/anaconda3/etc/profile.d/conda.sh
+
+# activate required conda environment
+conda activate climax
+
+# change to project root directory
+cd ~/git/climax
+
+# execute model training
+time python climax/main/downscale.py
-- 
GitLab