code: def __init__(self):
      super(ColoLinear, self).__init__()
      self._register_shard_params(["weight", "bias"])
  
imports: from colossalai.legacy.tensor import ComputePattern, ProcessGroup, ShardSpec
  from .colo_module import ColoModule
label_desc: Use image node_id 1 to create a new ColoLinear object from inherited base classes: ColoModule with example: obj = ColoLinear()