code: def __init__(self):
      self.conv = [
          ConvBlock(28, 28, 1),
          ConvBlock(28, 28, 128),
          ConvBlock(14, 14, 128),
      ]
      self.weight1 = Tensor.scaled_uniform(128, 10)
      self.weight2 = Tensor.scaled_uniform(128, 10)
  
imports: import sys
  import numpy
  from tinygrad.nn.state import get_parameters
  from tinygrad.tensor import Tensor
  from tinygrad.nn import BatchNorm2d, optim
  from tinygrad.helpers import getenv
  from extra.datasets import fetch_mnist
  from extra.augment import augment_img
  from extra.training import train, evaluate
label_desc: Use image node_id 1 to create a new BigConvNet object  with example: obj = BigConvNet()