Module: tf_agents.policies.policy_saver

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

POLICY_SPECS_PBTXT 'policy_specs.pbtxt'
absolute_import Instance of __future__._Feature
division Instance of __future__._Feature
print_function Instance of __future__._Feature