code: def train_loader_func(config, batch_size):
      train_loader, _, _ = load_dataset(config["data_dir"], batch_size)
      return train_loader
  
file_path: BigDL/python/orca/tutorial/pytorch/transfer_learning/train.py
imports: import os
  import numpy
  import matplotlib.pyplot
  import torch
  import torch.nn
  import torch.optim
  from torch.optim import lr_scheduler
  import torchvision
  from torchvision import datasets, transforms
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.pytorch import Estimator
  from bigdl.orca.learn.metrics import Accuracy
label_desc: Use image node_id 2 for calling a global function with example usage: train_loader_func(config, batch_size) and returns: train_loader