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

specs_from_collect_data_spec(...): Creates policy specs from specs loaded from disk.

Type Aliases

InputFnAndSpecType: The central part of internal API.

InputFnType: The central part of internal API.

Other Members

  • POLICY_SPECS_PBTXT = 'policy_specs.pbtxt'