Skip to content
Snippets Groups Projects
Commit 112f668d authored by Xuan Gu's avatar Xuan Gu
Browse files

Update file submit_benchmark_jobs.sh

parent c5c646c0
No related branches found
No related tags found
No related merge requests found
......@@ -2,13 +2,13 @@
set -e
export MODEL_NAME=nnunet_for_pytorch
export MODLE_VERSION=21.11.0
export MODLE_VERSION=latest
export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet
export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODEL_NAME}_${MODLE_VERSION}.sif
export GPU_RESERVATION=nodeimage
export TIME_RESERVATIOn=0-0:20:00
mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
benchmark_modes=("train" "predict")
# node_types=("thin" "fat")
node_types=("thin")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment