code: def __init__(self):
      super().__init__()
      self.conv = torch.nn.Conv1d(8, 16, 3)
      self.proj = torch.nn.Linear(32, 7)
  
file_path: megatron-lm/tests/unit_tests/dist_checkpointing/test_optimizer.py
imports: import numpy
  import torch
  from torch.optim import Adam
  from megatron.core import parallel_state
  from megatron.core.dist_checkpointing import ShardedTensor, save, load
  from megatron.core.dist_checkpointing.dict_utils import nested_values
  from megatron.core.dist_checkpointing.optimizer import get_param_id_to_sharded_param_map, optim_state_to_sharding_state
  from megatron.core.dist_checkpointing.utils import extract_sharded_tensors
  from tests.unit_tests.dist_checkpointing import TempNamedDir
  from tests.unit_tests.test_utilities import Utils
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: torch.nn with example: obj = Model()