From bef3bd7f9c59b7c1218250f3d52ed0a47555d027 Mon Sep 17 00:00:00 2001 From: James Byrne Date: Sat, 23 Mar 2024 07:22:46 +0000 Subject: [PATCH] Dev #38: sticking in some stubs for dawn use --- ensemble/predict.tmpl.yaml | 2 +- ensemble/template/dawn.sh | 5 +++++ ensemble/train.tmpl.yaml | 4 ++-- 3 files changed, 8 insertions(+), 3 deletions(-) create mode 100644 ensemble/template/dawn.sh 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