class_docstr: LeNet-5 Model with customizable hyper-parameters
docstr: Override ``Model.call`` to build LeNet-5 model.
docstr_tok: ["Override","``","Model.call","``","to","build","LeNet-5","model","."]
code: def call(self, x):
      """Override ``Model.call`` to build LeNet-5 model."""
      x = self.conv1(x)
      x = self.pool1(x)
      x = self.conv2(x)
      x = self.pool2(x)
      x = self.flatten(x)
      x = self.fc1(x)
      x = self.dropout(x)
      return self.fc2(x)
  
file_path: nni/examples/trials/mnist-tfv2/mnist.py
imports: import logging
  import tensorflow
  from tensorflow.keras import Model
  from tensorflow.keras.callbacks import Callback
  from tensorflow.keras.layers import Conv2D, Dense, Dropout, Flatten, MaxPool2D
  from tensorflow.keras.optimizers import Adam
  import nni
label_desc: Use image node_id 2 for calling the MnistModel obj's underlying member method code with example usage: obj.call(x) and returns: self