Rewrites the provided SavedModel.
tfx.components.trainer.rewriting.converters.rewrite_saved_model(
src: Text,
dst: Text,
rewriter_inst: tfx.components.trainer.rewriting.rewriter.BaseRewriter
,
dst_model_type: tfx.components.trainer.rewriting.rewriter.ModelType
= tfx.components.trainer.rewriting.rewriter.ModelType.SAVED_MODEL
)
Args |
src
|
location of the saved_model to rewrite.
|
dst
|
location of the rewritten saved_model.
|
rewriter_inst
|
the rewriter instance to invoke. Must inherit from
rewriter.BaseRewriter .
|
dst_model_type
|
the rewriter.ModelType of the destination model.
|