diff --git a/ensemble/predict.tmpl.yaml b/ensemble/predict.tmpl.yaml index f960174..55ad34d 100644 --- 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: someone@example.com 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 new file mode 100644 index 0000000..e73154b --- /dev/null +++ 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 index 0bed0bb..0a7e1c0 100644 --- 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: someone@example.com job_file: icenet_train.sh - cluster: gpu - nodes: 1 ntasks: NTASKS length: 4-00:00:00 maxruns: 5