def export_module_spec_with_checkpoint(module_spec,
                                       checkpoint_path,
                                       export_path,
                                       scope_prefix=""):
  """Exports given checkpoint as tfhub module with given spec."""

  # The main requirement is that it is possible to know how to map from
  # module variable name to checkpoint variable name.
  # This is trivial if the original code used variable scopes,
  # but can be messy if the variables to export are interwined
  # with variables not export.
  with tf.Graph().as_default():
    m = hub.Module(module_spec)
    assign_map = {
        scope_prefix + name: value for name, value in m.variable_map.items()
    }
    tf.train.init_from_checkpoint(checkpoint_path, assign_map)
    init_op = tf.initializers.global_variables()
    with tf.Session() as session:
      session.run(init_op)
      m.export(export_path, session)
def export_module_spec_with_checkpoint ( module_spec , checkpoint_path , export_path , scope_prefix = "" ) : with tf . Graph ( ) . as_default ( ) : m = hub . Module ( module_spec ) assign_map = { scope_prefix + name : value for name , value in m . variable_map . items ( ) } tf . train . init_from_checkpoint ( checkpoint_path , assign_map ) init_op = tf . initializers . global_variables ( ) with tf . Session ( ) as session : session . run ( init_op ) m . export ( export_path , session )
Exports given checkpoint as tfhub module with given spec.