A_Guide_to_Running_Tensorflow_Models_on_Android icon indicating copy to clipboard operation
A_Guide_to_Running_Tensorflow_Models_on_Android copied to clipboard

Optimized save method

Open JcmeLs opened this issue 8 years ago • 0 comments

This is not a issue,but a optimized save method.

  1. import dependency
from tensorflow.python.framework.graph_util import convert_variables_to_constants 
  1. Convert variables to constants and save
output_graph_def = convert_variables_to_constants(sess, sess.graph_def, output_node_names=
['output/outputs'])
    with tf.gfile.FastGFile('model/CTNModel.pb', mode='wb') as f:
        f.write(output_graph_def.SerializeToString())

JcmeLs avatar Jun 15 '17 01:06 JcmeLs