code: def sample_data(dataset, batch_size, image_size=4):
      dataset.resolution = image_size
      loader = DataLoader(
          dataset,
          shuffle=True,
          batch_size=batch_size,
          num_workers=1,
          last_batch="discard",
      )
      return loader
  
imports: import os
  import random
  import math
  import argparse
  import logging
  import os.path
  from io import BytesIO
  import lmdb
  import numpy
  from PIL import Image
  from tqdm import tqdm
  from mpl_toolkits.axes_grid1 import ImageGrid
  import matplotlib.pyplot
  import mxnet
  import mxnet.ndarray
  from mxnet import gluon, autograd
  from mxnet.gluon.data.vision import transforms
  from mxnet.gluon.data import DataLoader
  from model import StyledGenerator, Discriminator
label_desc: Use image node_id 3 for calling a global function with example usage: sample_data(dataset, batch_size, image_size) and returns: loader