View source on GitHub
|
A tff.program.ReleaseManager that releases values to a file system.
Inherits From: ReleaseManager
tff.program.SavedModelFileReleaseManager(
root_dir: Union[str, os.PathLike[str]], prefix: str = 'release_'
)
A tff.program.SavedModelFileReleaseManager is a utility for releasing values
from a federated program to a file system and is used to release values from
platform storage to customer storage in a federated program.
Values are released to the file system using the SavedModel (see
tf.saved_model) format. When the value is released, if the value is a value
reference or a structure containing value references, each value reference is
materialized. The value is then flattened and released using the SavedModel
format. The structure of the value is discarded.
See https://www.tensorflow.org/guide/saved_model for more information about the SavedModel format.
Args | |
|---|---|
root_dir
|
A path on the file system to save program state. If this path does not exist it will be created. |
prefix
|
A string to use as the prefix for filenames. |
Raises | |
|---|---|
ValueError
|
If root_dir is an empty string.
|
Methods
release
release(
value, type_signature, key
)
Releases value from a federated program.
| Args | |
|---|---|
value
|
A tff.program.ReleasableStructure to release.
|
type_signature
|
The tff.Type of value.
|
key
|
An integer used to reference the released value; key represents a
step in a federated program.
|