code: def build_whole_model():
      inputs = Input(shape=(28, 28, 1))
      x = Conv2D(32, 3, activation="relu")(inputs)
      x = Flatten()(x)
      x = Dense(128, activation="relu")(x)
      outputs = Dense(10)(x)
      return Model(
          inputs=inputs, outputs=outputs, name="vfl_whole_model"
      )
  
imports: from multiprocessing import Process
  import unittest
  import numpy
  import os
  from bigdl.ppml.fl import *
  from bigdl.ppml.fl.estimator import Estimator
  from bigdl.ppml.fl.nn.fl_server import FLServer
  from bigdl.ppml.fl.nn.tensorflow.utils import set_one_like_parameter
  from bigdl.ppml.fl.nn.fl_context import init_fl_context
  from bigdl.ppml.fl.nn.tensorflow.estimator import TensorflowEstimator
  import tensorflow
  from tensorflow.keras.layers import Dense, Flatten, Conv2D, InputLayer
  from tensorflow.keras import Model, Input
  import shutil
label_desc: Use image node_id 3 for calling a global function with example usage: build_whole_model() and returns: Model