ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.system.default.multiprocessing_core

Multiprocessing hooks for TF-Agents.


class StateSaver: Class for getting and setting global state.


enable_interactive_mode(...): Function that enables multiprocessing in interactive mode.

handle_main(...): Function that wraps the main function in a multiprocessing-friendly way.

handle_test_main(...): Function that wraps the test main in a multiprocessing-friendly way.