[Resnet50/pytorch] a GPU DataNode call gpu() function

NVIDIA / DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

13.59k stars 3.24k forks source link

[Resnet50/pytorch] a GPU DataNode call gpu() function #1419

Open tendar opened 2 months ago

tendar commented 2 months ago

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?