class_docstr: Dummy PyTorch model 
code: def __init__(self):
      super(CustomPytorchModel, self).__init__()
      self.cnn = nn.Sequential(
          nn.Conv2d(
              in_channels=3, out_channels=4, kernel_size=3, padding=1
          ),
      )
      self.dense = nn.Sequential(
          nn.Linear(in_features=16, out_features=32)
      )
  
file_path: maze/maze/test/perception/blocks/test_torch_model_block.py
imports: from typing import Dict
  import torch
  import torch.nn
  from maze.perception.blocks.general.torch_model_block import TorchModelBlock
  from maze.test.perception.perception_test_utils import build_multi_input_dict
label_desc: Use image node_id 1 to create a new CustomPytorchModel object from inherited base classes: nn with example: obj = CustomPytorchModel()