Basado en estoconvertir-entrenado-tensorflow-modelo-a-protobuf Estoy tratando de guardar / restaurar el gráfico TF sin éxito.

Aquí está el ahorrador:

with tf.Graph().as_default():
    variable_node = tf.Variable(1.0, name="variable_node")
    output_node = tf.mul(variable_node, 2.0, name="output_node")
    sess = tf.Session()
    init = tf.initialize_all_variables()
    sess.run(init)
    output = sess.run(output_node)
    tf.train.write_graph(sess.graph.as_graph_def(), summ_dir, 'model_00_g.pbtxt', as_text=True)
    #self.assertNear(2.0, output, 0.00001)
    saver = tf.train.Saver()
    saver.save(sess, saver_path)

que producemodel_00_g.pbtxt con descripción gráfica del texto. Más o menos copiar y pegar defreeze_graph_test.py.

Aquí está el lector:

with tf.Session() as sess:

    with tf.Graph().as_default():
        graph_def = tf.GraphDef()
        graph_path = '/mnt/code/test_00/log/2016-02-11.22-37-46/model_00_g.pbtxt'
        with open(graph_path, "rb") as f:
            proto_b = f.read()
            #print proto_b   # -> I can see it
            graph_def.ParseFromString(proto_b) # no luck..
            _ = tf.import_graph_def(graph_def, name="")

    print sess.graph_def

que falla engraph_def.ParseFromString() conDecodeError: Tag had invalid wire type.

Estoy en el contenedor acoplableb.gcr.io/tensorflow/tensorflow:latest-devel en caso de que haga alguna diferencia.