code: def __init__(self, input_size, output_size):
      super().__init__()
      self.l1 = nn.Linear(input_size, 5)
      self.relu = nn.ReLU()
      self.l2 = nn.Linear(5, output_size)
  
imports: import pytest
  import pytorch_lightning
  import torch
  import torch.nn
  import torch.nn.functional
  from sklearn.datasets import load_diabetes
  from torch.utils.data import Dataset
  from torchvision import transforms
  from torchvision.datasets import MNIST
  import nni
  import nni.nas.evaluator.pytorch.lightning
  from nni.nas.evaluator import FunctionalEvaluator, Evaluator
  from nni.nas.space import RawFormatModelSpace
label_desc: Use image node_id 1 to create a new FCNet object from inherited base classes: nn with example: obj = FCNet(input_size, output_size)