Module: tf_agents.policies.policy_saver
Stay organized with collections
Save and categorize content based on your preferences.
TF-Agents SavedModel API.
Classes
class PolicySaver: A PolicySaver allows you to save a tf_policy.Policy to SavedModel.
Functions
add_batch_dim(...)
rename_spec_with_nest_paths(...)
specs_from_collect_data_spec(...): Creates policy specs from specs loaded from disk.
Type Aliases
InputFnAndSpecType
InputFnType
Other Members |
|---|
|
POLICY_SPECS_PBTXT
|
'policy_specs.pbtxt'
|
|
absolute_import
|
Instance of __future__._Feature
|
|
division
|
Instance of __future__._Feature
|
|
print_function
|
Instance of __future__._Feature
|
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.
[null,null,["Last updated 2024-04-26 UTC."],[],[],null,["# Module: tf_agents.policies.policy_saver\n\n\u003cbr /\u003e\n\n|---------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/policies/policy_saver.py) |\n\nTF-Agents SavedModel API.\n\nClasses\n-------\n\n[`class PolicySaver`](../../tf_agents/policies/PolicySaver): A `PolicySaver` allows you to save a `tf_policy.Policy` to `SavedModel`.\n\nFunctions\n---------\n\n[`add_batch_dim(...)`](../../tf_agents/policies/policy_saver/add_batch_dim)\n\n[`rename_spec_with_nest_paths(...)`](../../tf_agents/policies/policy_saver/rename_spec_with_nest_paths)\n\n[`specs_from_collect_data_spec(...)`](../../tf_agents/policies/policy_saver/specs_from_collect_data_spec): Creates policy specs from specs loaded from disk.\n\nType Aliases\n------------\n\n[`InputFnAndSpecType`](../../tf_agents/policies/policy_saver/InputFnAndSpecType)\n\n[`InputFnType`](../../tf_agents/policies/policy_saver/InputFnType)\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Other Members ------------- ||\n|--------------------|-----------------------------------|\n| POLICY_SPECS_PBTXT | `'policy_specs.pbtxt'` |\n| absolute_import | Instance of `__future__._Feature` |\n| division | Instance of `__future__._Feature` |\n| print_function | Instance of `__future__._Feature` |\n\n\u003cbr /\u003e"]]
View source on GitHub