Module: tf_agents.policies.policy_saver

View source on GitHub

TF-Agents SavedModel API.

Classes

class PolicySaver: A PolicySaver allows you to save a tf_policy.Policy to SavedModel.

Other Members

  • COLLECT_POLICY_SPEC = 'collect_data_spec.pbtxt'