tf_agents.utils.common.function

Wrapper for tf.function with TF Agents-specific customizations.

tf_agents.utils.common.function(
    *args, **kwargs
)

Used in the tutorials
Checkpointer and PolicySaver

@common.function()
def my_eager_code(x, y):
  ...

Args
`*args`	Args for tf.function.
`**kwargs`	Keyword args for tf.function.

Returns
A tf.function wrapper.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.

Args