Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

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.

Functions

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

Other Members

  • POLICY_SPECS_PBTXT = 'policy_specs.pbtxt'