# Module: tf_agents.policies.policy_saver

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/policies/policy_saver.py) |

TF-Agents SavedModel API.

## Classes

[`class PolicySaver`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/PolicySaver): A `PolicySaver` allows you to save a `tf_policy.Policy` to `SavedModel`.

## Functions

[`add_batch_dim(...)`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/policy_saver/add_batch_dim)

[`rename_spec_with_nest_paths(...)`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/policy_saver/rename_spec_with_nest_paths)

[`specs_from_collect_data_spec(...)`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/policy_saver/specs_from_collect_data_spec): Creates policy specs from specs loaded from disk.

## Type Aliases

[`InputFnAndSpecType`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/policy_saver/InputFnAndSpecType)

[`InputFnType`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/policies/policy_saver/InputFnType)

<br />

<br />

<br />

| ## 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` |

<br />