code: def main(_):
      hparams = trainer_model_based_params.create_loop_hparams()
      problem_name = get_simulated_problem_name(hparams.game)
      world_model_dir = FLAGS.world_model_dir
      agent_model_dir = FLAGS.output_dir
      event_dir = FLAGS.output_dir
      epoch_data_dir = (
          FLAGS.data_dir
      )  # only required for initial frames
  
      trainer_model_based.train_agent(
          problem_name,
          agent_model_dir,
          event_dir,
          world_model_dir,
          epoch_data_dir,
          hparams,
          0,
          epoch=0,
          is_final_epoch=True,
      )
  
file_path: tensor2tensor/tensor2tensor/rl/trainer_model_based_agent_only.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from tensor2tensor.bin import t2t_trainer
  from tensor2tensor.data_generators import gym_env
  from tensor2tensor.rl import trainer_model_based
  from tensor2tensor.rl import trainer_model_based_params
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling a global function with example usage: main(_) without return types