code: def __iter__(self):
      stream = torch.cuda.Stream()
      first = True
  
      for next_input, next_target in self.loader:
          with torch.cuda.stream(stream):
              next_input = next_input.cuda(non_blocking=True)
              next_target = next_target.cuda(non_blocking=True)
              next_input = (
                  next_input.float().sub_(self.mean).div_(self.std)
              )
  
          if not first:
              yield input, target
          else:
              first = False
  
          torch.cuda.current_stream().wait_stream(stream)
          input = next_input
          target = next_target
  
      yield input, target
  
file_path: gen-efficientnet-pytorch/data/loader.py
imports: import torch
  import torch.utils.data
  from .transforms import *
label_desc: Use image node_id 2 for calling the PrefetchLoader obj's underlying member method code with example usage: obj.__iter__() without return types