code: def _toy_model_config(self, shard_size):
      config_dict = {
          "train_micro_batch_size_per_gpu": 2,
          "steps_per_print": 1,
          "optimizer": {
              "type": "Adam",
              "params": {
                  "lr": 0.00015,
                  "betas": [0.8, 0.999],
                  "eps": 1e-8,
                  "weight_decay": 3e-7,
              },
          },
          "fp16": {"enabled": True, "initial_scale_power": 8},
          "wall_clock_breakdown": True,
          "zero_optimization": {
              "stage": 3,
              "mics_shard_size": shard_size,
          },
      }
  
      hidden_dim = 10
      with deepspeed.zero.MiCS_Init(config_dict_or_path=config_dict):
          models = [
              SimpleModel(hidden_dim, empty_grad=False)
              for _ in range(2)
          ]
  
      return config_dict, hidden_dim, models
  
file_path: deepspeed/tests/unit/checkpoint/test_mics_optimizer.py
imports: import deepspeed
  from deepspeed.runtime.utils import required_torch_version
  from unit.common import DistributedTest
  from unit.simple_model import *
  from unit.checkpoint.common import *
  import pytest
label_desc: Use image node_id 1 for calling the TestMiCSCheckpoint obj's underlying member method code with example usage: obj._toy_model_config(shard_size) and returns: config_dict,  hidden_dim,  models