code: def __init__(self, nclass, colors, n):
      self.pre_conv = objax.nn.Sequential(
          [objax.nn.Conv2D(colors, n, k=3), objax.functional.leaky_relu]
      )
      self.block1 = simple_net_block(1 * n, 2 * n)
      self.block2 = simple_net_block(2 * n, 4 * n)
      self.post_conv = objax.nn.Conv2D(4 * n, nclass, k=3)
  
file_path: objax/examples/image_classification/mnist_cnn.py
imports: import os
  import numpy
  import tensorflow_datasets
  from tqdm import trange
  import objax
  from objax.util import EasyDict
label_desc: Use image node_id 1 to create a new SimpleNet object from inherited base classes: objax with example: obj = SimpleNet(nclass, colors, n)