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.
Type Aliases
InputFnAndSpecType: The central part of internal API.
InputFnType: The central part of internal API.
Other Members | |
|---|---|
| POLICY_SPECS_PBTXT |
'policy_specs.pbtxt'
|