View source on GitHub
|
Generates a decorator that sets default parameters for tf.function.
tf_agents.utils.common.set_default_tf_function_parameters(
*args, **kwargs
)
Args | |
|---|---|
*args
|
default arguments for the tf.function.
|
**kwargs
|
default keyword arguments for the tf.function.
|
Returns | |
|---|---|
Function decorator with preconfigured defaults for tf.function.
|