TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

hub.create_module_spec_from_saved_model

hub.create_module_spec_from_saved_model(
    saved_model_path,
    drop_collections=None
)

Experimental: Create a ModuleSpec out of a SavedModel.

Define a ModuleSpec from a SavedModel. Note that this is not guaranteed to work in all cases and it assumes the SavedModel has followed some conventions:

  • The serialized SaverDef can be ignored and instead can be reconstructed.
  • The init op and main op can be ignored and instead the module can be initialized by using the conventions followed by tf.train.MonitoredSession.

Note that the set of features supported can increase over time and have side effects that were not previously visible. The pattern followed to avoid surprises is forcing users to declare which features to ignore (even if they are not supported).

Note that this function creates a ModuleSpec that when exported exports a Module (based on a modified copy of the original SavedModel) and not a SavedModel.

Args:

  • saved_model_path: Directory with the SavedModel to use.
  • drop_collections: Additionally list of collection to drop.

Returns:

A ModuleSpec.