code: def __init__(self, in_size, hid_size, out_size):
      super().__init__()
      self.layers = nn.ModuleList()
      # 2-layer GraphSAGE-mean
      self.layers.append(SAGEConv(in_size, hid_size, "mean"))
      self.layers.append(SAGEConv(hid_size, out_size, "mean"))
      self.dropout = nn.Dropout(0.5)
      self.hid_size = hid_size
      self.out_size = out_size
  
file_path: cugraph/python/cugraph-dgl/examples/multi_trainer_MG_example/model.py
imports: import time
  import dgl
  import torch
  import torch.nn
  import torch.nn.functional
  import torchmetrics.functional
  from cugraph_dgl.nn import SAGEConv
  import tqdm
label_desc: Use image node_id 1 to create a new Sage object from inherited base classes: nn with example: obj = Sage(in_size, hid_size, out_size)