code: def get_data_loader(
      dataset_info,
      args,
      benchmark_config,
      model_specs,
      num_replicas=1,
      rank=0,
  ):
      return wikitext2_data.get_dataloaders(
          dataset_info,
          benchmark_config,
          model_specs,
          num_replicas,
          rank,
      )
  
file_path: fairscale/benchmarks/utils.py
imports: import argparse
  from functools import reduce
  import logging
  import operator
  import datasets.wikitext2_data
  from models import transformer_lm
  import numpy
  import torch
  from torch.optim import Adam
label_desc: Use image node_id 10 for calling a global function with example usage: get_data_loader(dataset_info, args, benchmark_config, model_specs, num_replicas, rank) and returns: wikitext2_data