From 3df5cdb45dac88081c65f36260b6686669ed33cc Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 16 Dec 2022 20:21:42 +0000
Subject: [PATCH] Slurm: finish writing for DeepLabV3+ rainfall

It isn't tested yet though, naturally.
---
 aimodel/slurm-TEST-deeplabv3p-rainfall.job | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/aimodel/slurm-TEST-deeplabv3p-rainfall.job b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
index e9291cb..a88d5de 100755
--- a/aimodel/slurm-TEST-deeplabv3p-rainfall.job
+++ b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
@@ -55,8 +55,17 @@ dir_output="output/$(date -u --rfc-3339=date)_${CODE}";
 
 echo -e ">>> Additional args: ${ARGS}";
 
-export PATH=$HOME/software/bin:$PATH;
-export 
+export PATH="$HOME/software/bin:$PATH";
+
+
+export IMAGE_SIZE;
+export BATCH_SIZE;
+export DATA_DIR_TF="$DIR_RAINFALL";
+export PATH_HEIGHTMAP;
+export PATH_COLOURMAP;
+export STEPS_PER_EPOCH;
+export DIR_OUTPUT="$dir_output";
+
 
 echo ">>> Installing requirements";
 conda run -n py38 pip install -r requirements.txt;