Attend the Women in ML Symposium on December 7 Register now

Module: tf_agents.drivers

Stay organized with collections Save and categorize content based on your preferences.

Drivers for running a policy in an environment.

Modules

driver module: Base class for drivers that takes steps in an environment.

dynamic_episode_driver module: A Driver that takes N episodes in the environment using a tf.while_loop.

dynamic_step_driver module: A Driver that takes N steps in the environment using a tf.while_loop.

py_driver module: A Driver that steps a python environment using a python policy.