@@ -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