Python train_agent 예제들

예제 #1

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,
                                    epoch=0,
                                    is_final_epoch=True)

예제 #2

파일: trainer_model_based_agent_only.py 프로젝트: qixiuai/tensor2tensor

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)