code: def __init__(self, data_size, hidden_size, mlp_size, num_layers):
      super().__init__()
  
      self._initial = MLP(
          1 + data_size, hidden_size, mlp_size, num_layers, tanh=False
      )
      self._func = DiscriminatorFunc(
          data_size, hidden_size, mlp_size, num_layers
      )
      self._readout = torch.nn.Linear(hidden_size, 1)
  
imports: import fire
  import matplotlib.pyplot
  import torch
  import torch.optim.swa_utils
  import torchcde
  import torchsde
  import tqdm
label_desc: Use image node_id 1 to create a new Discriminator object from inherited base classes: torch.nn with example: obj = Discriminator(data_size, hidden_size, mlp_size, num_layers)