Function that wraps the test main in a multiprocessing-friendly way.
tf_agents.system.default.multiprocessing_core.handle_test_main(
parent_main_fn, *args, **kwargs
)
This function additionally accepts an extra_state_savers kwarg;
users can provide a list of tf_agents.multiprocessing.StateSaver instances,
where a StateSaver tells multiprocessing how to store some global state
and how to restore it in the subprocess.
Args |
|---|
parent_main_fn
|
A callable.
|
*args
|
rgs for parent_main_fn.
|
**kwargs
|
kwargs for parent_main_fn. This may also include
extra_state_savers kwarg.
|
Returns |
|---|
Output of parent_main_fn.
|
View source on GitHub