Python train_evalの例

コード例 #1

ファイル: train_eval_clip_agent_atari.py プロジェクト: ymodak/agents

def main(_):
    tf.compat.v1.enable_resource_variables()
    logging.set_verbosity(logging.INFO)
    train_eval_clip_agent.train_eval(
        FLAGS.root_dir,
        tf_master=FLAGS.master,
        env_name=FLAGS.env_name,
        env_load_fn=suite_atari.load,
        replay_buffer_capacity=FLAGS.replay_buffer_capacity,
        num_environment_steps=FLAGS.num_environment_steps,
        num_parallel_environments=FLAGS.num_parallel_environments,
        num_epochs=FLAGS.num_epochs,
        collect_episodes_per_iteration=FLAGS.collect_episodes_per_iteration,
        num_eval_episodes=FLAGS.num_eval_episodes)

コード例 #2

ファイル: train_eval_clip_agent_random_py_env.py プロジェクト: zircote/agents

def main(_):
    tf.compat.v1.enable_resource_variables()
    if tf.executing_eagerly():
        # self.skipTest('b/123777119')  # Secondary bug: ('b/123775375')
        return
    logging.set_verbosity(logging.INFO)
    train_eval_clip_agent.train_eval(
        FLAGS.root_dir,
        tf_master=FLAGS.master,
        env_name=FLAGS.env_name,
        env_load_fn=env_load_fn,
        replay_buffer_capacity=FLAGS.replay_buffer_capacity,
        num_environment_steps=FLAGS.num_environment_steps,
        num_parallel_environments=FLAGS.num_parallel_environments,
        num_epochs=FLAGS.num_epochs,
        collect_episodes_per_iteration=FLAGS.collect_episodes_per_iteration,
        num_eval_episodes=FLAGS.num_eval_episodes,
        use_rnns=FLAGS.use_rnns)