code: def base_config():
      config_dict = {
          "train_batch_size": 1,
          "optimizer": {"type": "Adam", "params": {"lr": 0.00015}},
          "fp16": {"enabled": True},
      }
      return config_dict
  
imports: import torch
  import pytest
  import json
  import argparse
  from deepspeed.runtime.zero.config import DeepSpeedZeroConfig
  from unit.common import DistributedTest, get_test_path
  from unit.simple_model import SimpleModel, create_config_from_dict, random_dataloader
  import deepspeed.comm
  import deepspeed
  from deepspeed.runtime.config import DeepSpeedConfig, get_bfloat16_enabled
label_desc: Use image node_id 1 for calling a global function with example usage: base_config() and returns: config_dict