Michal Futrega

You can find the notebook with BraTS21 [here](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/nnUNet/notebooks/BraTS21.ipynb). Recently, we've released the notebook for BraTS22 which was ranked 2, you can find it [here](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/nnUNet/notebooks/BraTS22.ipynb).

Have you tried running with `--affinity disabled` or commenting the L32-33 in the `main.py`? (https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/nnUNet/main.py#L32). Another fix for torchmetrics is to upgrade pytorch lightning to 1.5.10 (there are issues with...

Have you tried running with `--affinity disabled` or commenting the L32-33 in the `main.py`? (https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/nnUNet/main.py#L32). Another fix for torchmetrics is to upgrade pytorch lightning to 1.5.10 (there are issues with...

Hi, I've run the command for 2 GPUs and it works fine for me: ``` oot@6e38dc6f86a4:/workspace/nnunet_pyt# python scripts/benchmark.py --mode train --gpus 2 --dim 3 --batch_size 2 --amp 387 training, 97...

Hi @sbmalik, do you run the notebook on xView2 dataset or using something different? In case you are using different dataset you might to modify some code, e.g. the metric...

Thanks for spotting that! I will push code with fix soon. In the meantime you can fix it locally by replacing [this line](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/nnUNet/data_loading/dali_loader.py#L45) to the `self.input_y = self.get_reader(kwargs["lbls"]) if kwargs["lbls"]...

@nv-kkudrynski this is already fixed.

jenkins

jenkins

Are you running notebook inside docker container? It looks like a dependency issue (running notebook with different versions of dependency). Please see https://ploomber.io/blog/notebook-to-docker/ for reference how to run Jupyter Notebook...