class_docstr: Wrapper class of Dataset class that performs multi-threaded data loading
docstr: Return a batch of data
docstr_tok: ["Return","a","batch","of","data"]
code: def __iter__(self):
      """Return a batch of data"""
      for i, data in enumerate(self.dataloader):
          if i * self.opt.batch_size >= self.opt.max_dataset_size:
              break
          yield data
  
imports: import sys
  import os
  import pathlib
  import logging
  import time
  import argparse
  from collections import namedtuple
  import numpy
  import torch
  import torch.utils.data
  import nni
  from nni.utils import merge_parameter
  from pix2pixlib.data.aligned_dataset import AlignedDataset
  from pix2pixlib.models.pix2pix_model import Pix2PixModel
  from base_params import get_base_params
label_desc: Use image node_id 4 for calling the CustomDatasetDataLoader obj's underlying member method code with example usage: obj.__iter__() without return types