code: def train_data_creator(config, batch_size):
      train_dataset = LinearDataset(
          2, 5, size=config.get("data_size", 1000)
      )
      train_loader = torch.utils.data.DataLoader(
          train_dataset, batch_size=batch_size
      )
      return train_loader
  
file_path: BigDL/python/orca/example/learn/horovod/pytorch_estimator.py
imports: import argparse
  import numpy
  import torch
  import torch.nn
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.pytorch import Estimator
label_desc: Use image node_id 4 for calling a global function with example usage: train_data_creator(config, batch_size) and returns: train_loader