Merge: [nnUNet/PyT] Fix DALI inference pipeline

Author: nv-kkudrynski

PyTorch/Segmentation/nnUNet/Dockerfile

@@ -5,7 +5,7 @@ RUN pip install nvidia-pyindex
 ADD ./requirements.txt .
 RUN pip install --disable-pip-version-check -r requirements.txt
 RUN pip install monai==0.8.0 --no-dependencies
-RUN yes | pip uninstall torchtext
+RUN pip uninstall -y torchtext
 
 RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
 RUN unzip -qq awscliv2.zip

PyTorch/Segmentation/nnUNet/data_loading/dali_loader.py

@@ -42,7 +42,7 @@ def __init__(self, batch_size, num_threads, device_id, **kwargs):
         self.patch_size = kwargs["patch_size"]
         self.load_to_gpu = kwargs["load_to_gpu"]
         self.input_x = self.get_reader(kwargs["imgs"])
-        self.input_y = self.get_reader(kwargs["lbls"]) if "lbls" in kwargs else None
+        self.input_y = self.get_reader(kwargs["lbls"]) if kwargs["lbls"] is not None else None
 
     def get_reader(self, data):
         return ops.readers.Numpy(

PyTorch/Segmentation/nnUNet/requirements.txt

@@ -5,4 +5,3 @@ pytorch-lightning==1.3.8
 scikit-learn==1.0
 scikit-image==0.18.3
 pynvml==11.0.0
-numpy==1.21.3