code: def create_model():
      return tf.keras.models.Sequential(
          [
              tf.keras.layers.Flatten(input_shape=(28, 28)),
              tf.keras.layers.Dense(512, activation="relu"),
              tf.keras.layers.Dropout(wandb.config.dropout),
              tf.keras.layers.Dense(10, activation="softmax"),
          ]
      )
  
file_path: wandb/tests/functional_tests/t0_main/mp/13-synctb-gradienttape.py
imports: import argparse
  import datetime
  import numpy
  import tensorflow
  import wandb
label_desc: Use image node_id 3 for calling a global function with example usage: create_model() and returns: tf