code: def save_model():
      print("Saving trained model")
      tf.io.gfile.makedirs(FLAGS.rundir + "/model")
      tf.compat.v1.train.Saver().save(
          sess, FLAGS.rundir + "/model/export"
      )
  
imports: import argparse
  import json
  import tensorflow
  from tensorflow.examples.tutorials.mnist import input_data
label_desc: Use image node_id 17 for calling a global function with example usage: save_model() without return types