[Resnet50/pytorch] a GPU DataNode call gpu() function #1419
Description
Related to Resnet50/pytorch
Describe the bug
source code link https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Classification/ConvNets/image_classification/dataloaders.py#L153
If here define the pipeline with param dali_cpu=Ture, then the output of images = self.res(images) is already a gpu DataNode. Is it OK call the gpu() function at LINE:153?