code: def __init__(self, hidden_dim, num_classes=10):
      super().__init__()
      self.layer1 = nn.Linear(hidden_dim, 2 * hidden_dim)
      self.layer2 = nn.Linear(2 * hidden_dim, 2 * hidden_dim)
      self.layer3 = nn.Linear(2 * hidden_dim, num_classes)
  
file_path: hivemind/tests/test_utils/custom_networks.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from hivemind.moe import register_expert_class
label_desc: Use image node_id 1 to create a new MultilayerPerceptron object from inherited base classes: nn with example: obj = MultilayerPerceptron(hidden_dim, num_classes)