code: def __init__(self):
      self.layers: List[Callable[[Tensor], Tensor]] = [
          nn.Conv2d(1, 32, 5),
          Tensor.relu,
          nn.Conv2d(32, 32, 5),
          Tensor.relu,
          nn.BatchNorm2d(32),
          Tensor.max_pool2d,
          nn.Conv2d(32, 64, 3),
          Tensor.relu,
          nn.Conv2d(64, 64, 3),
          Tensor.relu,
          nn.BatchNorm2d(64),
          Tensor.max_pool2d,
          lambda x: x.flatten(1),
          nn.Linear(576, 10),
      ]
  
imports: from typing import List, Callable
  from tinygrad import Tensor, TinyJit, nn, GlobalCounters
  from extra.datasets import fetch_mnist
  from tqdm import trange
label_desc: Use image node_id 1 to create a new Model object  with example: obj = Model()