Dev icenet-ai#38: sticking in some stubs for dawn use

JimCircadian · Mar 23, 2024 · bef3bd7 · bef3bd7
1 parent d2098e9
commit bef3bd7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/ensemble/predict.tmpl.yaml b/ensemble/predict.tmpl.yaml
@@ -14,6 +14,7 @@ ensemble:
       - ../../../processed
       - ../../../results
     mem:          224gb
+    cluster:      pvc
 
   pre_process:          []
   post_process:         []
@@ -24,7 +25,6 @@ ensemble:
     - icenet_predict.sh.j2
     email:        [email protected]
     job_file:     icenet_predict.sh
-    cluster:      short
     nodes:        1
     ntasks:       8
     length:       00:30:00

diff --git a/ensemble/template/dawn.sh b/ensemble/template/dawn.sh
@@ -0,0 +1,5 @@
+#!/usr/bin/env bash
+
+module purge
+module load default-dawn
+module load intelpython-conda
diff --git a/ensemble/train.tmpl.yaml b/ensemble/train.tmpl.yaml
@@ -17,6 +17,8 @@ ensemble:
       - ../../../results
     gpus:         1
     mem:          128gb
+    cluster:      pvc
+    nodes:        1
 
   pre_process:
     - name:   execute
@@ -31,8 +33,6 @@ ensemble:
     - icenet_train.sh.j2
     email:        [email protected]
     job_file:     icenet_train.sh
-    cluster:      gpu
-    nodes:        1
     ntasks:       NTASKS
     length:       4-00:00:00
     maxruns:      5