"""Binary for training TFModels with Estimator API."""

from absl import app
from absl import flags
import gin
from tensor2robot.utils import train_eval
import tensorflow.compat.v1 as tf


FLAGS = flags.FLAGS


def main(unused_argv):
  gin.parse_config_files_and_bindings(FLAGS.gin_configs, FLAGS.gin_bindings)
  train_eval.train_eval_model()


if __name__ == '__main__':
  tf.logging.set_verbosity(tf.logging.INFO)
  app.run(main)

"""Binary for training TFModels with Estimator API."""

from absl import app
from absl import flags
import gin
from tensor2robot.utils import train_eval
import tensorflow.compat.v1 as tf


FLAGS = flags.FLAGS


def main(unused_argv):
  gin.parse_config_files_and_bindings(
      FLAGS.gin_configs, FLAGS.gin_bindings, print_includes_and_imports=True)
  train_eval.train_eval_model()


if __name__ == '__main__':
  tf.logging.set_verbosity(tf.logging.INFO)
  app.run(main)

Print gin config includes and imports when starting training.
Print gin config includes and imports when starting training.

PiperOrigin-RevId: 342367820

  """Binary for training TFModels with Estimator API."""
  
  from absl import app
  from absl import flags
  import gin
  from tensor2robot.utils import train_eval
  import tensorflow.compat.v1 as tf
  
  
  FLAGS = flags.FLAGS
  
  
  def main(unused_argv):
-   gin.parse_config_files_and_bindings(FLAGS.gin_configs, FLAGS.gin_bindings)
+   gin.parse_config_files_and_bindings(
+       FLAGS.gin_configs, FLAGS.gin_bindings, print_includes_and_imports=True)
    train_eval.train_eval_model()
  
  
  if __name__ == '__main__':
    tf.logging.set_verbosity(tf.logging.INFO)
    app.run(main)
  
Print gin config includes and imports when starting training.
## Code Before:
"""Binary for training TFModels with Estimator API."""

from absl import app
from absl import flags
import gin
from tensor2robot.utils import train_eval
import tensorflow.compat.v1 as tf


FLAGS = flags.FLAGS


def main(unused_argv):
  gin.parse_config_files_and_bindings(FLAGS.gin_configs, FLAGS.gin_bindings)
  train_eval.train_eval_model()


if __name__ == '__main__':
  tf.logging.set_verbosity(tf.logging.INFO)
  app.run(main)

## Instruction:
Print gin config includes and imports when starting training.
## Code After:
"""Binary for training TFModels with Estimator API."""

from absl import app
from absl import flags
import gin
from tensor2robot.utils import train_eval
import tensorflow.compat.v1 as tf


FLAGS = flags.FLAGS


def main(unused_argv):
  gin.parse_config_files_and_bindings(
      FLAGS.gin_configs, FLAGS.gin_bindings, print_includes_and_imports=True)
  train_eval.train_eval_model()


if __name__ == '__main__':
  tf.logging.set_verbosity(tf.logging.INFO)
  app.run(main)
