class_docstr: Adapter that handles torch Dataloaders.
code: def get_dataset(self):
      def parse_dataloader_gen():
          self.current_step = 0
          for batch in iter(self._dataset):
              self.current_step += 1
              batch = map_structure(
                  lambda x: x.cpu().numpy(), list_to_tuple(batch)
              )
              yield batch
  
      return parse_dataloader_gen
  
file_path: elegy/elegy/data/torch_dataloader_adapter.py
imports: from jax._src.lax.lax import remaining
  from torch.utils.data import DataLoader
  from .data_adapter import DataAdapter
  from .utils import is_none_or_empty, list_to_tuple, map_structure
label_desc: Use image node_id 3 for calling the TorchDataLoaderAdapter obj's underlying member method code with example usage: obj.get_dataset() and returns: parse_dataloader_gen