code: def get_dataset_info(args):
      assert args.model_name == "lm"
      if args.use_synthetic_data:
          return wikitext2_data.get_synthetic_datasets()
      else:
          return wikitext2_data.get_real_datasets()
  
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 9 for calling a global function with example usage: get_dataset_info(args) and returns: wikitext2_data, wikitext2_data