code: def __init__(self):
      super(Network, self).__init__()
      self.conv1 = nn.Conv2d(1, 32, kernel_size=3)
      self.conv2 = nn.Conv2d(32, 64, kernel_size=3)
      self.fc1 = nn.Linear(1600, 128)
      self.fc2 = nn.Linear(128, 10)
  
file_path: pywick/tests/integration/fit_loader_simple/single_input_multi_target.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from torch.utils.data import DataLoader
  from pywick.modules import ModuleTrainer
  from pywick.datasets.TensorDataset import TensorDataset
  import os
  from torchvision import datasets
label_desc: Use image node_id 1 to create a new Network object from inherited base classes: nn with example: obj = Network()