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

tf_agents.utils.common.save_spec

View source on GitHub

Saves the given spec nest as a StructProto.

tf_agents.utils.common.save_spec(
    spec, file_path
)

Note: Currently this will convert BoundedTensorSpecs into regular TensorSpecs.

Args:

  • spec: A nested structure of TensorSpecs.
  • file_path: Path to save the encoded spec to.