View source on GitHub
|
This class exports the serving graph and checkpoints at the end.
Inherits From: Exporter
tf.estimator.FinalExporter(
name, serving_input_receiver_fn, assets_extra=None, as_text=False
)
This class performs a single export at the end of training.
Raises | |
|---|---|
ValueError
|
if any arguments is invalid. |
Attributes | |
|---|---|
name
|
Directory name.
A directory name under the export base directory where exports of
this type are written. Should not be |
Methods
export
export(
estimator, export_path, checkpoint_path, eval_result, is_the_final_export
)
Exports the given Estimator to a specific format.
| Args | |
|---|---|
estimator
|
the Estimator to export.
|
export_path
|
A string containing a directory where to write the export. |
checkpoint_path
|
The checkpoint path to export. |
eval_result
|
The output of Estimator.evaluate on this checkpoint.
|
is_the_final_export
|
This boolean is True when this is an export in the
end of training. It is False for the intermediate exports during the
training. When passing Exporter to tf.estimator.train_and_evaluate
is_the_final_export is always False if TrainSpec.max_steps is
None.
|
| Returns | |
|---|---|
The string path to the exported directory or None if export is skipped.
|
View source on GitHub