View source on GitHub |
Multiprocessing hooks for TF-Agents.
Classes
class StateSaver
: Class for getting and setting global state.
Functions
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.