use tmpdir

mwalmsley · Nov 19, 2023 · 5ac6be9 · 5ac6be9
1 parent 84eb547
commit 5ac6be9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/only_for_me/narval/gpu_split.py b/only_for_me/narval/gpu_split.py
@@ -48,6 +48,7 @@ def configure_optimizers(self):
 def main():
 
    logging.basicConfig(level=logging.INFO)
+   logging.warning('Script start')
 
    parser = argparse.ArgumentParser()
    parser.add_argument('--save-dir', dest='save_dir', type=str)
@@ -67,7 +68,7 @@ def main():
       save_dir = '/home/walml/repos/temp'
 
    else:
-      save_dir = '/project/def-bovy/walml/temp'
+      save_dir = os.environ['SLURM_TMPDIR']
 
    schema = schemas.decals_all_campaigns_ortho_schema
 

diff --git a/only_for_me/narval/gpu_split.sh b/only_for_me/narval/gpu_split.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=0:05:0  
+#SBATCH --time=0:15:0  
 #SBATCH --nodes=1
 #SBATCH --ntasks=2
 #SBATCH --ntasks-per-node=2
@@ -20,6 +20,7 @@ export NCCL_BLOCKING_WAIT=1  #Set this environment variable if you wish to use t
 # echo "r$SLURM_NODEID master: $MASTER_ADDR"
 # echo "r$SLURM_NODEID Launching python script"
 
+echo 'Running script'
 REPO_DIR=/project/def-bovy/walml/zoobot
 srun $PYTHON $REPO_DIR/only_for_me/narval/gpu_split.py --gpus 1