tf_agents.utils.common.set_default_tf_function_parameters

View source on GitHub

Generates a decorator that sets default parameters for tf.function.

*args default arguments for the tf.function.
**kwargs default keyword arguments for the tf.function.

Function decorator with preconfigured defaults for tf.function.