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

Update file README.md

parent 7320f48d
No related branches found
No related tags found
No related merge requests found
......@@ -13,6 +13,7 @@ cd ~/DeepLearningExamples/nnunet_for_pytorch_$VERSION/
```
- Build the nnU-Net PyTorch Docker container
Change the pytorch-lightning version to 1.5.10 to avoid the `from torchmetrics.utilities.data import get_num_classes as _get_num_classes` error.
```
docker build -t nnunet .
```
......@@ -67,7 +68,7 @@ The input arguments are:
We will average the benchmark performance over the iterations. The maximum usable (without a OOM error) batch size is 256 and 128 for single and multi-node, respectively.
```
cd cd Berzelius-nnU-Net-Benchmark && mkdir -p sbatch_out
cd Berzelius-nnU-Net-Benchmark && mkdir -p sbatch_out
bash scripts/benchmark_sbatch_submit.sh 2 1 8 1 128
```
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment