code: def get_weights(self):
      if isinstance(self.fc1.weight, ShardedTensor):
          weight1 = self.fc1.weight.local_tensor()
      else:
          weight1 = self.fc1.weight
  
      if isinstance(self.fc2.weight, ShardedTensor):
          weight2 = self.fc2.weight.local_tensor()
      else:
          weight2 = self.fc2.weight
  
      return (weight1, weight2)
  
file_path: pytorch/torch/testing/_internal/distributed/_shard/test_common.py
imports: import torch
  import torch.nn
  from torch.distributed._shard.sharded_tensor import ShardedTensor
label_desc: Use image node_id 3 for calling the SimpleMegatronLM obj's underlying member method code with example usage: obj.get_weights() and returns: weight1,  weight2