Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

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.