code: def _get_test_network(net=nn.Sequential()):
      net.add(
          nn.Dense(128, activation="relu", flatten=False),
          nn.Dense(64, activation="relu"),
          nn.Dense(10, activation="relu"),
      )
      return net
  
imports: import os
  import logging
  import sys
  import re
  import mxnet
  from common import TemporaryDirectory
  from mxnet import np
  from mxnet.gluon import nn, loss
  from mxnet.gluon.contrib.estimator import estimator, event_handler
  from mxnet.gluon.contrib.estimator.event_handler import LoggingHandler
  from mxnet.gluon.contrib.estimator.event_handler import ValidationHandler
  from mxnet.gluon.contrib.estimator import EpochEnd
  from mxnet.gluon.data.dataset import Dataset
label_desc: Use image node_id 1 for calling a global function with example usage: _get_test_network(net) and returns: net