diff --git a/README.md b/README.md index 1b39c5ff355891e8591677840ef3cccbc2eecab9..6226ff88614b9f17f9b18e89fef6f58b67b04106 100644 --- a/README.md +++ b/README.md @@ -31,6 +31,12 @@ docker push xuagu37/nvidia_nnu-net_for_pytorch:21.11.0 cd /proj/nsc/xuan/ngc git clone https://github.com/NVIDIA/DeepLearningExamples ``` +Run the script. You need to modify the script for e.g. the name of your reservation, number of nodes, batch_size, etc. +``` +cd /proj/nsc/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet +sbash benchmark_nnunet_pytorch_berzelius.sh +``` + <!-- #### For single node - Start an interactive session