Save the date! Google I/O returns May 18-20 Register now

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.