code: def main(unused_argv):
      tf.random.set_random_seed(20200823)
      np.random.seed(20200823)
  
      logging.info("=> Starting ...")
  
      # Select dataset.
      logging.info("=> Preparing datasets ...")
      input_fn = datasets.get_dataset("train", FLAGS)
  
      # Select model.
      logging.info("=> Creating {} model".format(FLAGS.model))
      model_fn = models.get_model(FLAGS)
  
      # Set up training.
      logging.info("=> Setting up training ...")
      run_config = tf_estimator.RunConfig(
          model_dir=FLAGS.train_dir,
          save_checkpoints_steps=FLAGS.save_every,
          save_summary_steps=FLAGS.summary_every,
          keep_checkpoint_max=None,
      )
      trainer = tf_estimator.Estimator(
          model_fn=model_fn,
          config=run_config,
      )
  
      # Start training.
      logging.info("=> Training ...")
      trainer.train(input_fn=input_fn, max_steps=FLAGS.max_steps)
  
imports: import numpy
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
  from tensorflow_graphics.projects.nasa.lib import datasets
  from tensorflow_graphics.projects.nasa.lib import models
  from tensorflow_graphics.projects.nasa.lib import utils
label_desc: Use image node_id 1 for calling a global function with example usage: main(unused_argv) without return types