code: def __init__(self, hiddens=128):
      super(MyModel, self).__init__()
      self.conv1 = Conv2D(32, 3, activation="relu")
      self.flatten = Flatten()
      self.d1 = Dense(hiddens, activation="relu")
      self.d2 = Dense(10, activation="softmax")
  
file_path: ray/python/ray/tune/examples/tf_mnist_example.py
imports: import argparse
  import os
  from filelock import FileLock
  from tensorflow.keras.layers import Dense, Flatten, Conv2D
  from tensorflow.keras import Model
  from tensorflow.keras.datasets.mnist import load_data
  from ray import train, tune
label_desc: Use image node_id 1 to create a new MyModel object from inherited base classes: Model with example: obj = MyModel(hiddens)